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) # 打印出导数
运行结果：
看完上述内容是否对您有帮助呢？如果还想对相关知识有进一步的了解或阅读更多相关文章，请关注亿速云行业资讯频道，感谢您对亿速云的支持。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
