torch.matmul的前后两个矩阵维度不同的小结

系列文章目录

本系列记录自己的代码学习知识

前言

一、torch.matmul()

torch.matmul()也是一种类似于矩阵相乘操作的tensor联乘操作。但是它可以利用python 中的广播机制，处理一些维度不同的tensor结构进行相乘操作。这也是该函数与torch.bmm()区别所在。

torch.matmul(input，other)#我们现以不同的名字区分前后两个矩阵

来源
作者：top_小酱油
链接：https://www.jianshu.com/p/e277f7fc67b3
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

二、详解解释

1.input维度比other大

例如：

n = ([1,2,3],[1,