一个上下文管理器(context manager)来连接需要计算梯度的函数和变量,方便求解同时也提升效率。
举个例子:计算y=x^2在x = 3时的导数:
import tensorflow as tf
x = tf.constant(3.0)
with tf.GradientTape() as g:
g.watch(x)
y = x * x
dy_dx = g.gradient(y, x) # y’ = 2*x = 2*3 = 6
tf.print(dy_dx)
一个上下文管理器(context manager)来连接需要计算梯度的函数和变量,方便求解同时也提升效率。
举个例子:计算y=x^2在x = 3时的导数:
import tensorflow as tf
x = tf.constant(3.0)
with tf.GradientTape() as g:
g.watch(x)
y = x * x
dy_dx = g.gradient(y, x) # y’ = 2*x = 2*3 = 6
tf.print(dy_dx)