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。

相关推荐
CS创新实验室1 分钟前
OpenAI GPT-5.5 技术深度报告
人工智能·gpt·大模型·llm
倒霉熊dd1 分钟前
Python 学习(第二部分:函数、模块与面向对象编程)
前端·数据库·python
dhashdoia2 分钟前
2026年深度体验:OpenAI Codex App如何重塑AI辅助编程工作流
人工智能·gpt·深度学习·claude
铁皮哥5 分钟前
【力扣题解】LeetCode 25. K 个一组翻转链表
java·数据结构·windows·python·算法·leetcode·链表
zhangfeng11338 分钟前
CodeBuddy ai对话框上面的git docs terminal Rulds 干嘛用的,以thinkphp fastadmin 为例,插件市场
人工智能·git·编程
dfsj6601115 分钟前
第六章:炼丹师的内功
人工智能
AIoT科技物语16 分钟前
包邮168元!无须编程,AI 驱动,ESP-Claw 物联网 OpenClaw 智能体套件,打通智能家居本地「感知、推理、决策」完整闭环
人工智能·物联网·智能家居
wzl2026121317 分钟前
从精细化运营视角,基于企销宝实现企微私域提速增效的技术实践
人工智能·自动化·企业微信
云天AI实战派17 分钟前
ChatGPT/AI 常见故障排查指南:从 Realtime API、Codex 到智能体的全流程修复手册
人工智能·chatgpt
ptc学习者19 分钟前
huggingface下载模型
人工智能