[Python知识库]微分12-TensorFlow语法可以结合python的控制流语法一起用

import tensorflow as tf



x = tf.constant(1.0)

v0 = tf.Variable(2.0)
v1 = tf.Variable(2.0)

with tf.GradientTape(persistent=True) as tape:
  tape.watch(x)
  if x > 0.0:
    result = v0
  else:
    result = v1**2


#  对目标里含有的变量求导（v1不含有，所以梯度为None）
dv0, dv1 = tape.gradient(result, [v0, v1])

print(dv0) # tf.Tensor(1.0, shape=(), dtype=float32)
print(dv1) # None


# 目标里中不具备的变量，肯定不可求导呀，比如这里的result对x
dx = tape.gradient(result, x)

print(dx) # None