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

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

Tensorflow中是否有内置功能或有效的方法？谢谢！