这篇文章主要介绍了PyTorch中torch.matmul()函数怎么使用的相关知识，内容详细易懂，操作简单快捷，具有一定借鉴价值，相信大家阅读完这篇PyTorch免费云主机域名中torch.matmul()函数怎么使用文章都会有所收获，下面我们一起来看看吧。pytorch中两个张量的乘法可以分为两种：两个张量对应元素相乘，在PyTorch中可以通过torch.mul函数（或*运算符）实现；两个张量矩阵相乘，在PyTorch中可以通过torch.matmul函数实现；torch.matmul(input, other) → Tensor
计算两个张量input和other的矩阵乘积
【注意】：matmul函数没有强制规定维度和大小，可以用利用广播机制进行不同维度的相乘操作。torch.matmul()也是一种类似于矩阵相乘操作的tensor连乘操作。但是它可以利用python中的广播机制，处理一些维度不同的tensor结构进行相乘操作。这也是该函数与torch.bmm()区别所在。若两个tensor都是一维的，则返回两个向量的点积运算结果：运行结果：tensor([1, 2]) tensor([3, 4])
tensor(11) torch.Size([])若两个tensor都是二维的，则返回两个矩阵的矩阵相乘结果：运行结果：tensor([[21, 24, 27],[47, 54, 61]]) torch.Size([2, 3])若input为一维，other为二维，则先将input的一维向量扩充到二维（维数前面插入长度为1的新维度），然后进行矩阵乘积，得到结果后再将此维度去掉，得到的与input的维度相同。运行结果：tensor([21, 24, 27]) torch.Size([3])【分析】：首先将x维度从(2)扩充为(,2)，然后将x(,2) 与y(2,3)进行相乘，得到(,3)，最后去掉一维部分，得到(3)若input为二维，other为一维，则先将other的一维向量扩充到二维（维数后面插入长度为1的新维度），然后进行矩阵乘积，得到结果后再将此维度去掉，得到的与other的维度相同。运行结果：tensor([ 50, 122])
torch.Size([2])【分析】：首先y维度从(3)扩充为(3,)，然后将x(2,3)与x(2,)进行相乘，得到(2,)，最后去掉一维部分，得到(2)【总结】：2.3和2.4基本类似，唯一不同的是2.3中一维向量和二维矩阵的乘积运算需要在一维向量前面插入长度为1的新维度（x为一维向量，y为二维矩阵）；2.4中二维矩阵和一维向量的乘积运算需要在一维向量后面插入长度为1的新维度（x为二维矩阵，y为一维向量）。关于“PyTorch中torch.matmul()函数怎么使用”这篇文章的内容就介绍到这里，感谢各位的阅读！相信大家对“PyTorch中torch.matmul()函数怎么使用”知识都有一定的了解，大家如果还想学习更多知识，欢迎关注百云主机行业资讯频道。
相关推荐: javascript怎么使用alert实现一个精美的弹窗
这篇文章主要讲解了“javascript怎么使用alert实现一个精美的弹窗”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“javascript怎么使用alert实现一个精美的弹窗”吧！其实最初使用alert还是…
免责声明：本站发布的图片视频文字，以转载和分享为主，文章观点不代表本站立场，本站不承担相关法律责任；如果涉及侵权请联系邮箱：360163164@qq.com举报，并提供相关证据，经查实将立刻删除涉嫌侵权内容。
微信扫一扫 
