我有一个神经网络模型，它为大约9个输入X输出大小约为4000的向量Y.我需要用输入X_1或X_2中的一个或两个来计算Y输出的偏导数。
我已经有了这些衍生物，我为X_1和X_2训练了两种不同的神经网络。它确实很好，但问题是衍生物不如计算Y的神经网络那么准确。
我希望有一种方法可以计算输出向量Y的导数到最终/优化神经网络的X输入之一，这样我就不需要为导数训练两个额外的神经网络。
有没有办法用autograd这样做？