基于python及pytorch中乘法的使用詳解
numpy中的乘法
A = np.array([[1, 2, 3], [2, 3, 4]]) B = np.array([[1, 0, 1], [2, 1, -1]]) C = np.array([[1, 0], [0, 1], [-1, 0]]) A * B : # 對應(yīng)位置相乘 np.array([[ 1, 0, 3], [ 4, 3, -4]]) A.dot(B) : # 矩陣乘法 ValueError: shapes (2,3) and (2,3) not aligned: 3 (dim 1) != 2 (dim 0) A.dot(C) : # 矩陣乘法 | < -- > np.dot(A, C) np.array([[-2, 2],[-2, 3]])
總結(jié) : 在numpy中,*表示為兩個數(shù)組對應(yīng)位置相乘; dot表示兩個數(shù)組進行矩陣乘法
pytorch中的乘法
A = torch.tensor([[1, 2, 3], [2, 3, 4]]) B = torch.tensor([[1, 0, 1], [2, 1, -1]]) C = torch.tensor([[1, 0], [0, 1], [-1, 0]]) # 矩陣乘法 torch.mm(mat1, mat2, out=None) <--> torch.matmul(mat1, mat2, out=None) eg : torch.mm(A, B) : RuntimeError: size mismatch, m1: [2 x 3], m2: [2 x 3] torch.mm(A, C) : tensor([[-2, 2], [-2, 3]]) torch.matmul(A, C) : tensor([[-2, 2], [-2, 3]]) # 點乘 torch.mul(mat1, mat2, out=None) eg : torch.mul(A, B) : tensor([[ 1, 0, 3], [ 4, 3, -4]]) torch.mul(A, C) : RuntimeError: The size of tensor a (3) must match the size of tensor b (2) at non-singleton dimension 1
總結(jié) : 在pytorch中,mul表示為兩個數(shù)組對應(yīng)位置相乘; mm和matmul表示兩個數(shù)組進行矩陣乘法
以上這篇基于python及pytorch中乘法的使用詳解就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python選擇網(wǎng)卡發(fā)包及接收數(shù)據(jù)包
今天小編就為大家分享一篇關(guān)于Python選擇網(wǎng)卡發(fā)包及接收數(shù)據(jù)包,小編覺得內(nèi)容挺不錯的,現(xiàn)在分享給大家,具有很好的參考價值,需要的朋友一起跟隨小編來看看吧2019-04-04解決Python 爬蟲URL中存在中文或特殊符號無法請求的問題
今天小編就為大家分享一篇解決Python 爬蟲URL中存在中文或特殊符號無法請求的問題。這種問題,初學者應(yīng)該都會遇到,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05python實現(xiàn)數(shù)通設(shè)備端口監(jiān)控示例
這篇文章主要介紹了python實現(xiàn)數(shù)通設(shè)備端口監(jiān)控示例,需要的朋友可以參考下2014-04-04python中使用paramiko模塊并實現(xiàn)遠程連接服務(wù)器執(zhí)行上傳下載功能
paramiko是用python語言寫的一個模塊,遵循SSH2協(xié)議,支持以加密和認證的方式,進行遠程服務(wù)器的連接。這篇文章主要介紹了python中使用paramiko模塊并實現(xiàn)遠程連接服務(wù)器執(zhí)行上傳下載功能,需要的朋友可以參考下2020-02-02python將Dataframe格式的數(shù)據(jù)寫入opengauss數(shù)據(jù)庫并查詢
這篇文章主要介紹了python將Dataframe格式的數(shù)據(jù)寫入opengauss數(shù)據(jù)庫并查詢,文章介紹詳細具有一定的參考價值,希望對你的學習有所幫助2022-04-04