pytorch | pytorch常用的乘法运算

2744557306

torch.mm : 用于两个矩阵（不包括向量）的乘法。如维度为(l,m)和(m,n)相乘
; Q9 |' {( I  e& q
' Y7 `( u$ [1 n1 X. Ttorch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
' `7 D6 B8 T1 o/ e: k$ \+ t' B
7 ^5 O7 n" k5 }7 _0 |  Storch.mul : 用于两个同维度矩阵的逐像素点相乘（点乘）。如维度为(l,m)和(l,m)相乘
, g! \. k9 h5 ?8 ^
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】
2 Z$ G2 e* f6 H7 z+ Z
@运算符 : 其作用类似于torch.matmul。
' g7 B5 T1 j4 P6 y5 @' j, A
4 l8 |$ @- p8 ^( C/ {' N*运算符 : 其作用类似于torch.mul。
+ V! x) f, K+ l* K& B/ ]
  X. i7 K6 i- A$ s" N# M1 H* T
# u6 G  Q5 a; c9 W