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中如何自动求梯度
新闻来源:http://azwzsj.com/article/jspogc.html