QQ登录

只需要一步,快速开始

 注册地址  找回密码
查看: 1902|回复: 0
打印 上一主题 下一主题

pytorch | pytorch常用的乘法运算

[复制链接]
字体大小: 正常 放大

1175

主题

4

听众

2810

积分

该用户从未签到

跳转到指定楼层
1#
发表于 2024-3-29 17:40 |只看该作者 |倒序浏览
|招呼Ta 关注Ta
torch.mm : 用于两个矩阵(不包括向量)的乘法。如维度为(l,m)和(m,n)相乘
" C! q( D; W; B( |3 T" q
; M* k' y! }1 I, }/ ?9 xtorch.bmm : 用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘9 y! g) y0 `. g3 ^- a* B3 n
3 U1 N; w' p* K2 S# S0 n( t' V3 e
torch.mul : 用于两个同维度矩阵的逐像素点相乘(点乘)。如维度为(l,m)和(l,m)相乘) w0 x: V$ M; K) _. G4 j/ q1 n7 v) W, M
% \* E& E( d, K7 Y7 h6 C* S6 {
torch.mv : 用于矩阵和向量之间的乘法(矩阵在前,向量在后)。如维度为(l,m)和(m)相乘,结果的维度为(l)。
8 d7 e1 u/ h. @$ C2 h- H
1 u" \9 S4 _, H# ?* K7 h/ ]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 C' i3 M" G& v8 s& P
7 a4 R+ \2 ]* X' V3 h5 z@运算符 : 其作用类似于torch.matmul。7 E8 @1 b$ X& I0 R% h  M

( M5 _2 a( H: U$ K5 o5 b*运算符 : 其作用类似于torch.mul。6 o. I- C8 z4 z' n$ M

5 L( H0 G2 f  m% J, A; o! F
, ?- B. j% v5 H- n3 a  c) j
7 n$ T& B. Y  \2 d. U4 Q/ H# G$ Y
zan
转播转播0 分享淘帖0 分享分享0 收藏收藏0 支持支持0 反对反对0 微信微信
您需要登录后才可以回帖 登录 | 注册地址

qq
收缩
  • 电话咨询

  • 04714969085
fastpost

关于我们| 联系我们| 诚征英才| 对外合作| 产品服务| QQ

手机版|Archiver| |繁體中文 手机客户端  

蒙公网安备 15010502000194号

Powered by Discuz! X2.5   © 2001-2013 数学建模网-数学中国 ( 蒙ICP备14002410号-3 蒙BBS备-0002号 )     论坛法律顾问:王兆丰

GMT+8, 2025-7-11 06:17 , Processed in 0.638260 second(s), 50 queries .

回顶部