这里总结一下pytorch常用的乘法运算以及相关的运算符(@
、*
)。
总结放前面:
**torch.mm :**用于两个矩阵(不包括向量)的乘法。如维度为(l,m)和(m,n)相乘
**torch.bmm :**用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
**torch.mul :**用于两个同维度矩阵的逐像素点相乘(点乘)。如维度为(l,m)和(l,m)相乘
**torch.mv :**用于矩阵和向量之间的乘法(矩阵在前,向量在后)。如维度为(l,m)和(m)相乘,结果的维度为(l)。
**torch.matmul :**用于两个张量(后两维满足矩阵乘法的维度)相乘或者是矩阵与向量间的乘法,因为其具有广播机制(broadcasting,自动补充维度)。如维度为(b,l,m)和(b,m,n);(l,m)和(b,m,n);(b,c,l,m)和(b,c,m,n);(l,m)和(m)相乘等。【其作用包含torch.mm、torch.bmm和torch.mv】
@运算符 : 其作用类似于torch.matmul。
*运算符 : 其作用类似于torch.mul。