Tensorflow2.0中如何自动求梯度

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

10余年建站经验, 成都网站设计、网站建设、外贸网站建设客户的见证与正确选择。创新互联建站提供完善的营销型网页建站明细报价表。后期开发更加便捷高效,我们致力于追求更美、更快、更规范。

对于给定函数: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)  # 打印出导数

运行结果:
Tensorflow2.0中如何自动求梯度

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


网站栏目:Tensorflow2.0中如何自动求梯度
路径分享:http://scyanting.com/article/jspogc.html