数学建模社区-数学中国
标题:
pytorch | pytorch常用的乘法运算
[打印本页]
作者:
2744557306
时间:
2024-3-29 17:40
标题:
pytorch | pytorch常用的乘法运算
torch.mm : 用于两个矩阵（不包括向量）的乘法。如维度为(l,m)和(m,n)相乘
: C5 T6 E1 K4 T
; q2 J r) y. E. w
torch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
7 j- d( l/ [7 l5 R, ^8 A; X
4 i5 f8 c9 R% ?
torch.mul : 用于两个同维度矩阵的逐像素点相乘（点乘）。如维度为(l,m)和(l,m)相乘
4 {; A q5 Z* m! Y
5 V# y2 L' q5 C
torch.mv : 用于矩阵和向量之间的乘法（矩阵在前，向量在后）。如维度为(l,m)和(m)相乘，结果的维度为(l)。
( G3 N% d$ s3 B. ]* y. y, `3 A
1 V0 p' P' X; h! X' ?% b3 S& } B
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】
+ w: M& k# v. V$ ^, k/ U
) K2 R/ m, e# Z' U3 T5 N; i% w' Q& m
@运算符 : 其作用类似于torch.matmul。
3 p W/ Z5 S7 h8 e( m, t6 E
4 ~; c* K7 m5 ]) V& Z$ w3 A2 t
*运算符 : 其作用类似于torch.mul。
! P. D- }3 q" k+ b( t% ?1 K
2 Q' ^8 j j1 N- T
2 F) u2 {8 S q& N, ]
7 Y# [; p$ _* ]0 t5 C4 o0 [" N4 }
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5