求梯度个人Tips

x = torch.tensor([[-1., 1.], [1., -1.]], requires_grad = True)
z = x.pow(2).sum()
z.backward()
print(x.grad)

'''输出
tensor([[-2.,  2.],
        [ 2., -2.]])
'''

• requires_grad = True
• backward的作用对象必须是一个标量
• 浮点数 或者 complex 类型的 tensor 支持梯度