tensorflow - 如何使用 GradientTape 计算二阶梯度？

Question

我正在尝试计算以下函数的偏导数，您能否建议这是否是使用 TensorFlow 计算偏导数的正确方法？

f(x,y,z) = xy+z pow 2 + 1/x

x = tf.Variable(2, name='x')
y = tf.Variable(1, name='y')
z = tf.constant(4, name='z')

with tf.GradientTape(persistent=True) as t:
  t.watch(x)
  f = x*y + tf.pow(z,2) + tf.divide(1,x)

g = tape.gradient(f, [x,y,z])

print(g)