1. 點乘，對應元素相乘，不求和

import torcha = torch.Tensor([[1,2], [3,4], [5,6]]) b1 = a.mul(a)// b2=a*a b1 Out[79]: tensor([[ 1., 4.],[ 9., 16.],[25., 36.]]) b2 Out[80]: tensor([[ 1., 4.],[ 9., 16.],[25., 36.]])

?以上兩種方法都可以表示點乘。

2.矩陣相乘

• 要求：前列=后行；

b2 = a.mm(a.t()) b3= a.matmul(a.t())

得到結果

tensor([[ 5., 11., 17.],[11., 25., 39.],[17., 39., 61.]]

以上兩種方法都可以

總結

以上是生活随笔為你收集整理的Pytorch——对应点相乘和矩阵相乘的全部內容，希望文章能夠幫你解決所遇到的問題。

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