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。

相关推荐
roman_日积跬步-终至千里12 小时前
【强化学习基础(5)】策略搜索与学徒学习:从专家行为中学习加速学习过程
人工智能
梁正雄13 小时前
2、Python流程控制
开发语言·python
Eric.Lee202114 小时前
ubuntu 安装 Miniconda
linux·运维·python·ubuntu·miniconda
无心水14 小时前
【Python实战进阶】1、Python高手养成指南:四阶段突破法从入门到架构师
开发语言·python·django·matplotlib·gil·python实战进阶·python工程化实战进阶
杭州泽沃电子科技有限公司14 小时前
在线监测:为医药精细化工奠定安全、合规与质量基石
运维·人工智能·物联网·安全·智能监测
GIS数据转换器14 小时前
GIS+大模型助力安全风险精细化管理
大数据·网络·人工智能·安全·无人机
李剑一14 小时前
Python学习笔记1
python
OJAC11114 小时前
AI跨界潮:金融精英与应届生正涌入人工智能领域
人工智能·金融
机器之心14 小时前
Adam的稳+Muon的快?华为诺亚开源ROOT破解大模型训练「既要又要」的两难困境
人工智能·openai
可观测性用观测云14 小时前
观测云 MCP Server 接入和使用最佳实践
人工智能