- Python 深度学习
- 吕云翔 刘卓然 关捷雄等编著
- 84字
- 2025-02-28 02:11:37
4.5 PyTorch的自动微分Autograd
当Tensor的requires_grad属性设置为True时,PyTorch的torch.autograd会自动追踪它的计算轨迹。当需要计算微分时,只需要对最终计算结果的Tensor调用backward方法,中间所有计算结点的微分就会被保存在grad属性中。


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