INNER CODE UNIT · Python

custom_polynomial

nfmcclure/tensorflow_cookbook · 01_Introduction/05_Declaring_Operations/05_operations.py:35

def custom_polynomial(x_val):
    # Return 3x^2 - x + 10
    return tf.subtract(3 * tf.square(x_val), x_val) + 10

print(sess.run(custom_polynomial(11)))

# What should we get with list comprehension
expected_output = [3*x*x-x+10 for x in test_nums]
print(expected_output)

# TensorFlow custom function output
for num in test_nums:
    print(sess.run(custom_polynomial(num)))

View source record →