Home / TensorFlow / Automatic differentiation with GradientTape
Example: Scalar
- Evaluate \(y = x^2\) at \(x = 10\).
- Find \(\dv*{y}{x}\) at said \(x\).
import tensorflow as tf
x = tf.constant(10.0)
with tf.GradientTape() as tape:
tape.watch(x)
y = x ** 2
dx = tape.gradient(y, x)
f"y = {y}, dy/dx = {dx}"
y = 100.0, dy/dx = 20.0