代码
import torch
a = torch.tensor([1.,2.,3.,4.], requires_grad=True)
b = a ** 2
print(b)
b.mean().backward()
print(a.grad)
输出
tensor([ 1., 4., 9., 16.], grad_fn=<PowBackward0>)
tensor([0.5000, 1.0000, 1.5000, 2.0000])
import torch
a = torch.tensor([1.,2.,3.,4.], requires_grad=True)
b = a ** 2
print(b)
b.mean().backward()
print(a.grad)
tensor([ 1., 4., 9., 16.], grad_fn=<PowBackward0>)
tensor([0.5000, 1.0000, 1.5000, 2.0000])