梯度计算
# 自动求导总结
x = torch.ones(3,3, requires_grad=True)
y = 9 * x ** 2
out = y.mean() # 计算矩阵的平均梯度
out.backward() # 前向计算梯度
# optimizer.step() # 反向更新梯度
print(x.grad)
输出
tensor([[2., 2., 2.],
[2., 2., 2.],
[2., 2., 2.]])