
資料內(nèi)容:
1. torch
torch模塊是torch的核心模塊,包含了PyTorch的大部分基礎(chǔ)操作,比如類型轉(zhuǎn)換,隨機數(shù)生成,矩陣運算,等等。
torch.from_numpy:從numpy的 ndarray中創(chuàng)建一個tensor對象。
torch.tensor:這個函數(shù)相當于是從數(shù)據(jù)源創(chuàng)建一個新的tensor,不需要任何參數(shù),可以從 list、array、tuple等等構(gòu)造出tensor對象。
torch.rand:隨機生成tensor。
2.Autograd
Autograd模塊是自動梯度計算模塊,主要用來計算當前節(jié)點對應(yīng)輸入變量的梯度。
torch.autograd.Function:這個函數(shù)用來定義自定義的 autograd 處理函數(shù),有前向傳播forward和反向傳播backward,可以自己定義backward 實現(xiàn)自定義的梯度算法。
torch.autograd.Variable:用這個變量可以計算變量的梯度,它本質(zhì)上是一個tensor對象,不同的是它支持自動求導。