PyTorch矩阵乘法
总结:pytorch和numpy中,星号*都表示矩阵点对点相乘;matmul都表示矩阵乘法。
===========================================================================
PyTorch中,a.mul(b)表示矩阵a和b点对点乘(即相同位置的元素相乘),‘*’也表示矩阵点对点相乘。a.mm(b)表示矩阵乘法。另外torch.matmul(a, b)以及a@b也都可以表示a和b的矩阵乘积。


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

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


浙公网安备 33010602011771号