torch中的几种乘法:torch.mm, torch.mul, torch.matmul &&矩阵相乘和矩阵点乘
amp 几种 矩阵 乘法 MM torch 相乘
2023-09-11 14:22:51 时间
一、点乘
点乘都是broadcast的,可以用torch.mul(a, b)实现,也可以直接用*实现。
矩阵点乘,就是矩阵各个对应元素相乘,要求矩阵必须维数相等,即MxN维矩阵乘以MxN维矩阵 。
二、矩阵乘
当a, b维度不一致时,会自动填充到相同维度相点乘。
矩阵相乘有torch.mm和torch.matmul两个函数。其中前一个是针对二维矩阵,后一个是高维。当torch.mm用于大于二维时将报错。
矩阵相乘,就是矩阵的乘法操作,要求左矩阵的列和右矩阵的行数要一样,即MN维矩阵乘以和NY维矩阵
相关文章
- Python&JS宏 实现保留样式合并表格后拆分
- 树状数组区间加法&区间求和操作
- 聊聊高并发(十九)理解并发编程的几种"性" -- 可见性,有序性,原子性
- MySQL 窗口函数 & MySQL8新特性
- spring & java 面试
- 《软件建模与设计: UML、用例、模式和软件体系结构》一一第2章 Software Modeling & Design: UML, Use Cases, Patterns, & Software Architectures
- jquery版本号升级不兼容的问题:$("input").attr("value")功能发生改变
- 扩展一个boot的插件—tooltip&做一个基于boot的表达验证
- 【图像处理】——改变图像的大小(降采样&重采样)下采样和上采样
- 学习笔记(27):Python网络编程&并发编程-GIL与多线程