PyTorch矩阵乘法

总结:pytorch和numpy中,星号*都表示矩阵点对点相乘;matmul都表示矩阵乘法。

===========================================================================

PyTorch中,a.mul(b)表示矩阵a和b点对点乘(即相同位置的元素相乘),‘*’也表示矩阵点对点相乘。a.mm(b)表示矩阵乘法。另外torch.matmul(a, b)也可以表示a和b的矩阵乘积。

对于a.mul(b),b也可以是标量:

对于.mul()和‘*’,维数不同也可以相乘:

 

posted @ 2020-03-27 11:22  Picassooo  阅读(1944)  评论(0编辑  收藏  举报