Можно ли использовать Autograd для вычисления производной нейронной сети по одному из ее входов?

Question

У меня есть модель нейронной сети, которая выводит вектор Y размером приблизительно 4000 для примерно 9 входов X. Мне нужно вычислить частную производную выхода Y с одним или двумя из входов X_1 или X_2.

У меня уже есть эти производные, и я обучил две разные нейронные сети для X_1 и X_2.Это хорошо, но проблема в том, что производные не так точны, как нейронная сеть, которая вычисляет Y.

Я надеюсь, что есть способ вычислить производные выходного вектора Y для одного извходы в X из окончательной / оптимизированной нейронной сети, так что мне не нужно будет обучать две дополнительные нейронные сети для производных.

Есть ли способ сделать это с помощью autograd?

Answer 1

Я не уверен, что вы имеете в виду, когда говорите, что вы подготовили отдельные сети для своих входов, но вы можете получить градиенты, используя TensorFlow tf.gradients op.