pytorch总结—自动求梯度

对函数求梯度(gradient)。PyTorch提供的autograd 包能够根据输⼊和前向传播过程自动构建计算图,并执⾏行反向传播。  Tensor 是这个包的核心类,如果将其属性 .requires_grad 设置为 True ,它将开始追 踪(track)在其上的所有操作(这样就可以利利⽤用链式法则进⾏行行梯度传播了了)。完成计算后,可以调 ⽤用 .backward() 来完成所有梯度计算。
相关文章
相关标签/搜索