PyTorch矩阵乘法

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

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

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

原文地址:https://www.cnblogs.com/picassooo/p/12580129.html