Python知識分享網(wǎng) - 專業(yè)的Python學習網(wǎng)站 學Python,上Python222
pytorch常用函數(shù)手冊 PDF 下載
匿名網(wǎng)友發(fā)布于:2025-10-23 09:42:26
(侵權(quán)舉報)
(假如點擊沒反應(yīng),多刷新兩次就OK!)

pytorch常用函數(shù)手冊 PDF 下載 圖1

 

 

資料內(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對象,不同的是它支持自動求導。