Tensorflow tf.GradientTape
Tensorflow tf.GradientTape
先来看一段代码
x = tf.constant(3.0)
with tf.GradientTape() as g:
g.watch(x)
y = x * x
dy_dx = g.gradient(y, x) # output 6.0 = 2x
可以发现 GradientTape 可以用来计算函数的导数
------------------------------分割线------------------------------
再来看一段代码
原创
2020-08-21 06:33:40 ·
232 阅读 ·
0 评论