pytorch | pytorch常用的乘法运算

这里总结一下pytorch常用的乘法运算以及相关的运算符(@*)。

总结放前面:

**torch.mm :**用于两个矩阵(不包括向量)的乘法。如维度为(l,m)和(m,n)相乘

**torch.bmm :**用于带batch的三维向量的乘法。如维度为(b,l,m)和(b,m,n)相乘

**torch.mul :**用于两个同维度矩阵的逐像素点相乘(点乘)。如维度为(l,m)和(l,m)相乘

**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.mmtorch.bmm和torch.mv

@运算符 : 其作用类似于torch.matmul。

*运算符 : 其作用类似于torch.mul。

相关推荐
渡我白衣几秒前
多路转接模型与select
人工智能·深度学习·websocket·网络协议·机器学习·网络安全·信息与通信
AIsdhuang3 分钟前
2025 AI培训权威推荐榜:深度评测与趋势前瞻
大数据·人工智能·python
dagouaofei4 分钟前
AI制作年终总结PPT零基础可用
人工智能·python·powerpoint
卿雪5 分钟前
MySQL【存储引擎】:InnoDB、MyISAM、Memory...
java·数据库·python·sql·mysql·golang
Q_Q5110082855 分钟前
python+django/flask+vue的基于文学创作的社交论坛系统
spring boot·python·django·flask·node.js·php
九河云7 分钟前
新能源汽车充电桩数字化:充电效率 AI 调控与运维服务云管理平台实践
运维·人工智能·汽车
咚咚王者8 分钟前
人工智能之数据分析 Pandas:第四章 常用函数
人工智能·数据分析·pandas
菩提树下的凡夫11 分钟前
Yolov11的空标注负样本技术在模型训练中的应用
人工智能·深度学习·yolo
Q_Q51100828512 分钟前
python+django/flask网红酒店预定系统
spring boot·python·django·flask·node.js·php
夕小瑶12 分钟前
DeepSeek V3.2的隐藏更新,却意外暴露了MiniMax
人工智能