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。

相关推荐
Memene摸鱼日报16 分钟前
「Memene 摸鱼日报 2025.9.16」OpenAI 推出 GPT-5-Codex 编程模型,xAI 发布 Grok 4 Fast
人工智能·aigc
xiaohouzi11223319 分钟前
OpenCV的cv2.VideoCapture如何加GStreamer后端
人工智能·opencv·计算机视觉
用户1252055970821 分钟前
解决Stable Diffusion WebUI训练嵌入式模型报错问题
人工智能
用户83562907805122 分钟前
从手动编辑到代码生成:Python 助你高效创建 Word 文档
后端·python
Juchecar24 分钟前
一文讲清 nn.LayerNorm 层归一化
人工智能
martinzh24 分钟前
RAG系统大脑调教指南:模型选择、提示设计与质量控保一本通
人工智能
小关会打代码25 分钟前
计算机视觉案例分享之答题卡识别
人工智能·计算机视觉
Juchecar25 分钟前
一文讲清 nn.Linear 线性变换
人工智能
c8i27 分钟前
python中类的基本结构、特殊属性于MRO理解
python
Se7en25842 分钟前
使用 NVIDIA Dynamo 部署 PD 分离推理服务
人工智能