数学建模社区-数学中国
标题:
pytorch | pytorch常用的乘法运算
[打印本页]
作者:
2744557306
时间:
2024-3-29 17:40
标题:
pytorch | pytorch常用的乘法运算
torch.mm : 用于两个矩阵（不包括向量）的乘法。如维度为(l,m)和(m,n)相乘
, f) A: S9 Q5 R
8 B1 S* h+ y: W/ R( m# t' M
torch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
5 E* Z3 |% ^( G) O
* w3 M! U6 ?2 T& k7 f/ W
torch.mul : 用于两个同维度矩阵的逐像素点相乘（点乘）。如维度为(l,m)和(l,m)相乘
, M% i" P& R' I) z% V8 M: l
W) X1 b7 V5 V& R/ K m
torch.mv : 用于矩阵和向量之间的乘法（矩阵在前，向量在后）。如维度为(l,m)和(m)相乘，结果的维度为(l)。
" F9 Q5 j; i- Y2 p( G* A
8 T, S: B. ~7 _) Y& F! g# r4 Y
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】
6 H! O+ w8 O: w6 z
2 M" q# ~- L8 n6 p% _( K$ @
@运算符 : 其作用类似于torch.matmul。
4 h' _: A& b2 z5 T9 _- I
4 D% k2 U' Q1 l! u: J/ }# { F
*运算符 : 其作用类似于torch.mul。
`9 Q. }( A- o. e/ q4 i9 L4 N9 j' ^& |
3 F: }) E c6 h. z4 O3 {& e
1 ^( E8 z7 A1 P9 Q0 a4 O
# c5 R! n8 t! x' {! s* f
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5