计算图
非标量调用backward()求梯度
x = torch.arange(4.0)
x.requires_grad = True
y = x * x
方法1:通过y.sum().backward()调用反向传播
y.sum().backward()
print(x.grad)
x = torch.arange(4.0)
x.requires_grad = True
y = x * x
y.sum().backward()
print(x.grad)