pytorch總結—自動求梯度

對函數求梯度(gradient)。PyTorch提供的autograd 包能夠根據輸⼊和前向傳播過程自動構建計算圖,並執⾏行反向傳播。  Tensor 是這個包的核心類,如果將其屬性 .requires_grad 設置爲 True ,它將開始追 蹤(track)在其上的所有操作(這樣就可以利利⽤用鏈式法則進⾏行行梯度傳播了了)。完成計算後,可以調 ⽤用 .backward() 來完成所有梯度計算。
相關文章
相關標籤/搜索