tensorflow--自动求导机制tf.GradientTape

一个上下文管理器(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)

猜你喜欢

转载自blog.csdn.net/fgg1234567890/article/details/115190846