pytorch:torch.mm()和torch.matmul()的使用
如下所示:
torch.mm(mat1, mat2, out=None) → Tensor torch.matmul(mat1, mat2, out=None) → Tensor
對矩陣mat1和mat2進(jìn)行相乘。 如果mat1 是一個n×m張量,mat2 是一個 m×p 張量,將會輸出一個 n×p 張量out。
參數(shù) :
mat1 (Tensor) – 第一個相乘矩陣
mat2 (Tensor) – 第二個相乘矩陣
out (Tensor, optional) – 輸出張量
代碼示例:
import torch a=torch.randn(2,3) b=torch.randn(3,2) print(torch.mm(a,b)) print(torch.matmul(a,b))
以上這篇pytorch:torch.mm()和torch.matmul()的使用就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Django項目定期自動清除過期session的2種方法實例
如果用戶主動退出,session會自動清除,如果沒有退出就一直保留,記錄數(shù)越來越大,要定時清理沒用的session,下面這篇文章主要給大家介紹了關(guān)于Django項目定期自動清除過期session的2種方法,需要的朋友可以參考下2022-08-08Python實現(xiàn)統(tǒng)計單詞出現(xiàn)的個數(shù)
這篇文章主要介紹了Python實現(xiàn)統(tǒng)計單詞出現(xiàn)的個數(shù),本文給出了實現(xiàn)代碼以及使用方法,需要的朋友可以參考下2015-05-05Python?Serial串口的簡單數(shù)據(jù)收發(fā)方式
這篇文章主要介紹了Python?Serial串口的簡單數(shù)據(jù)收發(fā)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09Python數(shù)據(jù)結(jié)構(gòu)之棧、隊列及二叉樹定義與用法淺析
這篇文章主要介紹了Python數(shù)據(jù)結(jié)構(gòu)之棧、隊列及二叉樹定義與用法,結(jié)合具體實例形式分析了Python數(shù)據(jù)結(jié)構(gòu)中棧、隊列及二叉樹的定義與使用相關(guān)操作技巧,需要的朋友可以參考下2018-12-12輕量級的Web框架Flask 中模塊化應(yīng)用的實現(xiàn)
說到flask的模塊化,大家可能第一時間想到的都是藍(lán)圖,今天我們不討論藍(lán)圖,先從0.2版本中的Module類的實現(xiàn)講起2017-09-09