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。

相关推荐
美林数据Tempodata2 分钟前
“双新”指引,AI驱动:工业数智应用生产性实践创新
大数据·人工智能·物联网·实践中心建设·金基地建设
电科_银尘9 分钟前
【大语言模型】-- 私有化部署
人工智能·语言模型·自然语言处理
惊讶的猫1 小时前
LSTM论文解读
开发语言·python
测试老哥2 小时前
软件测试之单元测试知识总结
自动化测试·软件测试·python·测试工具·职场和发展·单元测试·测试用例
翔云 OCR API2 小时前
人工智能驱动下的OCR API技术演进与实践应用
人工智能·ocr
buvsvdp50059ac2 小时前
如何在VSCode中设置Python解释器?
ide·vscode·python
南方者2 小时前
重磅升级!文心 ERNIE-5.0 新一代原生全模态大模型,这你都不认可它吗?!
人工智能·aigc
njxiejing2 小时前
Python进度条工具tqdm的安装与使用
开发语言·python
庄周迷蝴蝶2 小时前
旋转位置编码(Rotary Position Embedding,RoPE)
人工智能·机器学习
xier_ran2 小时前
深度学习:RMSprop 优化算法详解
人工智能·深度学习·算法