Differentiate the model input using torch.autograd.grad.
使用torch可以自动实现对网络参数的梯度计算,然而有时需要用到网络对输入变量的求导。该过程可通过torch.autograd.grad()
方法实现。
import torch.autograd as autogradautograd.grad(outputs, inputs, grad_outputs=None, retain_graph=None, create_graph=
来源地址:https://blog.csdn.net/universsky2015/article/details/131388825