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])