是否可以使用Autograd来计算神经网络输出相对于其输入之一的导数？

Question

我有一个神经网络模型，可以为大约9个输入X输出大小为4000的向量Y。我需要用输入X_1或X_2中的一个或两个来计算Y输出的偏导数。 / p>

我已经有了这些导数，并且已经为X_1和X_2训练了两个不同的神经网络。效果很好，但问题是导数不如计算Y的神经网络准确。

我希望有一种方法可以计算出输出矢量Y到来自最终确定/优化的神经网络的X中输入之一的导数，这样我就不需要为导数训练两个额外的神经网络。

是否可以使用autograd做到这一点？

Answer 1

我不确定您说要为输入训练单独的网络时是什么意思，但是可以使用TensorFlow tf.gradients op。

来获得渐变。