1 點乘

tensor與標量點乘的結果是每個元素乘以這個標量的值

tensor與行向量點乘的結果是每一列乘以對應的值

tensor與列向量點乘的結果是每一行乘以對應的值

矩陣進行點乘，就是對應位的乘積

點乘還有廣播的特點（也是對應位的乘積）

2?mul（與點乘相同）

tensor與標量mul的結果是每個元素乘以這個標量的值

tensor與行向量mul的結果是每一列乘以對應的值

tensor與列向量點乘的結果是每一行乘以對應的值

矩陣進行mul，就是對應位的乘積

、

mul也有廣播的特點（也是對應位的乘積）

3 mm 矩陣乘法

4 matmul 矩陣乘法

a,b矩陣同3中的a,b矩陣

矩陣乘法的廣播版本

5 bmm batch的乘法

計算兩個tensor的矩陣乘法。

對于torch.bmm(a,b)中的兩個參數，tensor a 的size為(b,h,w),tensor b的size為(b,w,h)。

注意兩個tensor的維度必須為3.

比三維多，比三維少的tensor維度都不行。

二維的情況：

四維的情況

四維矩陣相乘，可以用matmul實現

bmm使用（作用理論上和matmul一樣？甚至matmul功能還更強大一些？歡迎討論！）

總結

以上是生活随笔為你收集整理的pytorch笔记：pytorch的乘法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。