tensorflow 自动求导
Example:
import tensorflow as tf
# 创建 4 个张量,并赋值
a = tf.constant(1.)
b = tf.constant(2.)
c = tf.constant(3.)
w = tf.constant(4.)
with tf.GradientTape() as tape:# 构建梯度环境
tape.watch([w]) # 将 w 加入梯度跟踪列表
# 构建计算过程,函数表达式
y = a * w**2 + b * w + c
# 自动求导
[dy_dw] = tape.gradient(y, [w])
print(dy_dw) # 打印出导数
"""
tf.Tensor(10.0, shape=(), dtype=float32)
"""
因上求缘,果上努力~~~~ 作者:图神经网络,转载请注明原文链接:https://www.cnblogs.com/BlairGrowing/p/15939593.html