pytorch | pytorch常用的乘法运算

2744557306

torch.mm : 用于两个矩阵（不包括向量）的乘法。如维度为(l,m)和(m,n)相乘
; p" K7 H8 g4 w: z" c3 _
! [+ p+ {' ], m( a/ etorch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
# W% {. m$ \" ^7 H" W! D' h  x
( w9 w; u7 Z, L$ Storch.mul : 用于两个同维度矩阵的逐像素点相乘（点乘）。如维度为(l,m)和(l,m)相乘
# S: X/ K2 b$ G$ Z! M
8 ^0 r6 f8 s- P# b9 J  @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】
5 N7 P+ b0 k3 c9 f( T$ Q0 U
, [0 c+ N) N- ~! B7 G@运算符 : 其作用类似于torch.matmul。

*运算符 : 其作用类似于torch.mul。
! n9 P, a+ j; [
3 e, U6 `1 x6 {! P: d3 t
+ @3 r0 _7 y7 e4 i7 P7 I
" x3 Q0 |, u, i' l7 b