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

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

 

 

資料內(nèi)容:

 

1. torch

torch模塊是torch的核心模塊,包含了PyTorch的大部分基礎(chǔ)操作,比如類型轉(zhuǎn)換,隨機(jī)數(shù)生成,矩陣運(yùn)算,等等。

torch.from_numpy:從numpy的 ndarray中創(chuàng)建一個(gè)tensor對(duì)象。

torch.tensor:這個(gè)函數(shù)相當(dāng)于是從數(shù)據(jù)源創(chuàng)建一個(gè)新的tensor,不需要任何參數(shù),可以從 list、array、tuple等等構(gòu)造出tensor對(duì)象。

torch.rand:隨機(jī)生成tensor。

 

2.Autograd

Autograd模塊是自動(dòng)梯度計(jì)算模塊,主要用來(lái)計(jì)算當(dāng)前節(jié)點(diǎn)對(duì)應(yīng)輸入變量的梯度。

torch.autograd.Function:這個(gè)函數(shù)用來(lái)定義自定義的 autograd 處理函數(shù),有前向傳播forward和反向傳播backward,可以自己定義backward 實(shí)現(xiàn)自定義的梯度算法。

torch.autograd.Variable:用這個(gè)變量可以計(jì)算變量的梯度,它本質(zhì)上是一個(gè)tensor對(duì)象,不同的是它支持自動(dòng)求導(dǎo)。