（注意：这不是关于反向传播的问题。）我正在尝试在GPU上使用PyTorch张量代替Numpy数组求解非线性PDE。我想计算任意张量的偏导数，类似于中心有限差分numpy.gradient函数的作用。我有其他解决此问题的方法，但是由于我已经在使用PyTorch，所以我想知道是否有可能使用autograd模块（或者通常是任何其他自动分化模块）来执行此操作。
我创建了numpy.gradient函数的张量兼容版本-运行速度快得多。但是，也许有更优雅的方法可以做到这一点。我找不到其他任何资料可以解决这个问题，无论是表明这是可能的还是不可能的。也许这反映了我对自动分化算法的无知。