分类：autograd.grad函数

2021
06-23

pytorch 中autograd.grad()函数的用法说明

我们在用神经网络求解PDE时，经常要用到输出值对输入变量（不是Weights和Biases）求导；在训练WGAN-GP时，也会用到网络对输入变量的求导。以上两种需求，均可以用pytorch中的autograd.grad()函数实现。autograd.grad(outputs,inputs,grad_outputs=None,retain_graph=None,create_graph=False,only_inputs=True,allow_unused=False)outputs:求导的因变量（需要求导的函数）inputs:求导的自变量grad_outputs:如果ou... 继续阅读 >