Tensorflow2.0中如何自动求梯度

Tensorflow2.0中如何自动求梯度，很多新手对此不是很清楚，为了帮助大家解决这个难题，下面小编将为大家详细讲解，有这方面需求的人可以来学习下，希望你能有所收获。

对于给定函数：y(w)=aw^2+bw+c
数学求导得：dy/dw=2aw+b
那么，（a,b,c,w）=(1,2,3,4)处的导数，dy/dw=2 * 1 *4 + 2=10
而在Tensorflow2.0中，梯度可以自动求取。具体代码如下：

import tensorflow as tf

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])  # 函数y对w求导print(dy_dw)  # 打印出导数

运行结果：

看完上述内容是否对您有帮助呢？如果还想对相关知识有进一步的了解或阅读更多相关文章，请关注亿速云行业资讯频道，感谢您对亿速云的支持。