python-3.x - 如何在内部计算 pytorch 中的 autograd.grad？

Question

我试图在 pytorch 中获得毕业生。因此，我将其编码如下：

x = torch.tensor([1., 2., 3.], requires_grad=True)
y = x * x

torch.autograd.grad(y, x, torch.tensor([2., 2., 2.])

输出：

tensor([4., 8., 12.])

内部是如何计算的？

我不知道为什么要在 torch.autograd.grad() 中指定“grad_outputs”作为参数以及它是如何计算的。

标签： python-3.xpytorch