在Tensorflow中计算jacobian矩阵

在Tensorflow中计算jacobian矩阵

假设

X

和

Y

是Tensorflow张量，并且

Y

取决于

X

：

from tensorflow.python.ops.parallel_for.gradients import jacobianJ=jacobian(Y,X)

结果具有形状，

Y.shape + X.shape

并提供的每个元素

Y

相对于的每个元素的偏导数

X

。