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)  # 打印出导数



  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
 

运行结果：

文章来源: ai-wx.blog.csdn.net，作者：AI 菌，版权归原作者所有，如需转载，请联系作者。

原文链接：ai-wx.blog.csdn.net/article/details/104685737