tensorflow-gpu2.0之自動導數

import tensorflow as tf
'''
梯度：導數或偏導數
1.在什麼點的導數：在點（a,b,c,w）=(1,2,3,4)點的導數
2.梯度環境
    對誰求導：      對w求導
    函數：          y = a*w**2+b*w+c

以上三條是自動導數的必要信息
'''
##1.什麼點求導，須要建立定點張量
a = tf.constant(1.)
b = tf.constant(3.)
c = tf.constant(6.)
w = tf.constant(2.)
##2.求導環境，對誰求導
with tf.GradientTape() as tape: ##構建剃度環境
    tape.watch([w])             ##對w求導，[w]
    y = a*w**2+b*w+c            ##函數
##求導
[dy_dw]=tape.gradient(y,[w])
print(dy_dw)

輸出：tf.Tensor(7.0, shape=(), dtype=float32)

說明：

1.常量的建立（張量）使用tf.constant

2.剃度環境：tf.GradientTape()

3.對誰求導：tape.watch([w])