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。

相关推荐
Blossom.118几秒前
基于Mamba-2的实时销量预测系统:如何用选择性状态空间干掉Transformer的O(n²)噩梦
人工智能·python·深度学习·react.js·机器学习·设计模式·transformer
Mintopia1 分钟前
AIGC 技术标准制定:Web 行业协同的必要性与难点
人工智能·aigc·trae
Wise玩转AI3 分钟前
Day 26|智能体的“伦理与安全边界”
人工智能·python·安全·ai·chatgpt·ai智能体
极速learner4 分钟前
n8n本地安装的两种方法:小白入门大白话版本
人工智能·prompt
_codemonster5 分钟前
深度学习实战(基于pytroch)系列(三十八)门控循环单元(GRU)从零开始实现
人工智能·深度学习·gru
yang)5 分钟前
如何处理DAC的sinc滚降
人工智能
霍格沃兹测试开发学社-小明7 分钟前
自动化测试报告样式终极对比:HTMLTestRunner vs BeautifulReport vs HTMLReport vs Allure
人工智能
腾飞开源8 分钟前
07_Spring AI 干货笔记之提示词
人工智能·提示词·提示词工程·角色分配·模板渲染·spring ai·令牌机制
梦里不知身是客1111 分钟前
帆软的图标类型介绍
python·信息可视化·数据分析
h***381811 分钟前
使用python进行PostgreSQL 数据库连接
数据库·python·postgresql