Python 深度学习
上QQ阅读APP看书,第一时间看更新

4.5 PyTorch的自动微分Autograd

当Tensor的requires_grad属性设置为True时,PyTorch的torch.autograd会自动追踪它的计算轨迹。当需要计算微分时,只需要对最终计算结果的Tensor调用backward方法,中间所有计算结点的微分就会被保存在grad属性中。