数学建模社区-数学中国
标题:
pytorch | pytorch常用的乘法运算
[打印本页]
作者:
2744557306
时间:
2024-3-29 17:40
标题:
pytorch | pytorch常用的乘法运算
torch.mm : 用于两个矩阵（不包括向量）的乘法。如维度为(l,m)和(m,n)相乘
5 i4 L8 B, J' H8 u$ Q
- D% J/ x: Y6 q! N/ J; _
torch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘
' Z; |3 R+ s& F" a0 Z
5 U" j: y% A" Z% M4 {
torch.mul : 用于两个同维度矩阵的逐像素点相乘（点乘）。如维度为(l,m)和(l,m)相乘
4 T' _7 G. H8 Q, y5 A! j$ E
# L% ^6 V* G9 Q+ _
torch.mv : 用于矩阵和向量之间的乘法（矩阵在前，向量在后）。如维度为(l,m)和(m)相乘，结果的维度为(l)。
5 x. Q$ I( \, `; y7 @; `0 e
5 t+ n4 b; v2 _. r4 u' I* k g$ C
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】
$ S8 S% S8 L# |: C* x' W4 o8 n
7 l, h. s. d! v9 t: g- H
@运算符 : 其作用类似于torch.matmul。
6 C5 t7 z. I) e5 _2 A, ^; O R, f: |6 z
8 z' v6 L) t: \7 N
*运算符 : 其作用类似于torch.mul。
9 {$ @ w* Z5 i, p j
* U, E/ b0 [* m5 `% o
- w3 B5 L+ ?. o- q0 r6 m9 g; |! f
% C& y9 c& @9 \* ~! w! S& U; H
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5