Tensorflow 1.12:计算一个张量中的元素相对于另一个张量中的元素的导数

我有一个b×T×n张量Z和一个n×m矩阵X,我希望相对于X中共享相同索引的每个元素计算Z中每个元素的偏导数沿尺寸为n的轴。因此,我的输出将是一个b×T×n×m张量。

Tensorflow中是否有内置功能或有效的方法?谢谢!

评论