torch.matmul()用法介绍

torch.mm()

文章共322字 · 阅读需要大约2分钟

一键AI生成摘要，助你高效阅读

问答

明日何其多_

162969人浏览 · 2020-04-27 10:35:02

明日何其多_ · 2020-04-27 10:35:02 发布

torch.matmul是tensor的乘法，输入可以是高维的。
当输入都是二维时，就是普通的矩阵乘法，和tensor.mm函数用法相同。
在这里插入图片描述

当输入有多维时，把多出的一维作为batch提出来，其他部分做矩阵乘法。
在这里插入图片描述
下面看一个两个都是3维的例子。

将b的第0维1broadcast成2提出来，后两维做矩阵乘法即可。
再看一个复杂一点的，是官网的例子。

首先把a的第0维2作为batch提出来，则a和b都可看作三维。再把a的1broadcast成5，提取公因式5。（这样说虽然不严谨，但是便于理解。）然后a剩下(3,4)，b剩下(4,2)，做矩阵乘法得到(3,2)。

GitCode 开源社区

旨在为数千万中国开发者提供一个无缝且高效的云端环境，以支持学习、使用和贡献开源项目。

更多推荐

【目标检测】目标检测的一些常用神经网络模型及方法

我的阶段性总结????文章目录1.概述1.2 目标检测的任务1.3 目标检测的分类2.R-CNN系列2.1 [R-CNN（Region with CNN features）](https://arxiv.org/pdf/1311.2524.pdf)2.2 [Fast R-CNN](https://www.cv-foundation.org/openaccess/content_iccv_2015/