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。

相关推荐
Ybaocheng1 分钟前
大模型第一章
人工智能·机器学习·语言模型
2501_916008892 分钟前
Python抓包HTTPS详解:Wireshark、Fiddler、Charles等工具使用教程
python·ios·小程序·https·uni-app·wireshark·iphone
Gitpchy5 分钟前
Day 58 经典时序模型2
python
梵得儿SHI8 分钟前
(第一篇)Spring AI 核心技术攻坚:RAG 全流程落地指南|从理论到实战构建本地知识库问答系统
人工智能·spring·大模型落地·增强生成(rag)技术·大模型存在的知识滞后·大模型存在的知识幻觉·提升回答可信度
张彦峰ZYF9 分钟前
AI赋能原则3解读思考:可得性时代-AI 正在重写人类能力结构的未来
人工智能·ai·ai赋能与落地
CoovallyAIHub9 分钟前
AI 项目如何避免“烂尾”?怎么选择才能让AI项目长期奔跑?
人工智能·算法·计算机视觉
hudawei99611 分钟前
词嵌入中语料库矩阵和句子矩阵是怎样的?
人工智能·ai·自然语言处理·词嵌入·word embedding·词向量·语义理解
Mxsoft61911 分钟前
某次数据分析偏差,发现时区设置错,修正时间戳救场!
人工智能
2301_7951672013 分钟前
Python 高手编程系列一十五:使用 __new __()方法覆写实例创建过程
开发语言·网络·python
努力也学不会java13 分钟前
【docker】Docker Register(镜像仓库)
运维·人工智能·机器学习·docker·容器