Pytorch – 使用torch.matmul()替换torch.einsum(‘bhxyd,md->bhxym’,(a,b))算子模式
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Pytorch – 使用torch.matmul()替换torch.einsum(‘bhxyd,md->bhxym’,(a,b))算子模式
原文链接:https://www.stubbornhuang.com/2068/
发布于:2022年04月01日 9:37:40
修改于:2022年06月07日 16:23:29
在本文中,我将介绍如何使用普通算子matmul对torch.einsum('bhxyd,md->bhxym',(a,b))的算子模式进行替代。之前在另一篇文章:https://www.stubbornhuang.com/2065/中我已经详细介绍了pytorch的matmul方法,以及如何使用torch.matmul()替换torch.einsum('nctw,cd->ndtw',(a,b))模式,感兴趣的朋友可以看一下。
1 使用torch.matmul()替换torch.einsum('bhxyd,md->bhxym',(a,b))算子模式
torch.einsum('bhxyd,md->bhxym',(a,b))
的意思是将第一个维度为bhxyd
的张量Tensor a与第二个维度为md
的张量Tensor b进行batch矩阵相乘得到维度为bhxym
的结果张量Tensor。
这个einsum的算子模式的目的就是batch矩阵相乘,而这是可以通过matmul通过改写替换的。
首先我们将第二个张量b从md
调整为dm
,这个时候第一个张量a的最后一个维度与调整后b的维度一致,这样就满足matmul的计算规则。然后使用torch.matmul()将两个Tensor进行batch矩阵相乘得到结果Tensor,而结果Tensor维度为bhxym
,而这个tensor就是我们需要的Tensor。
详细的替换代码和示例代码如下:
(1)请购买正版资源,支持原作者和出版社;
(2)资源收集自互联网,仅供自我学习,请在下载后24小时内删除该资源,如下载者将此资源用于其他非法用途,本站不承担任何法律责任;如有侵权,请立即联系我,马上删除!
(3)此捐赠下载方式无需注册+登录账号,当然如果您喜欢本站也可注册账号;
(4)所支付的款项为捐赠本站款项,而并非付费下载所付款项,本站会在您捐赠本站后显示隐藏内容,请知悉,先非常感谢您对本站的捐赠;
(5)本站支付使用Payjs,该第三方支付方式为微信官方授权的服务商,无支付风险,详情请查看:https://payjs.cn;
(6)请尽量不要使用手机浏览器以及电脑浏览器的无痕模式进行支付操作,以免造成支付成功但未显示隐藏内容。
(7)点击立即购买按钮后弹出选择支付方式页面,选择任意支付方式点击后会跳转到新的二维码支付页面,支付成功后会弹出支付成功的提示框,点击提示框会自动刷新页面显示隐藏内容;另外请不要重复支付,避免财产损失;
(8)在2021年7月27日前,本站采用注册+登录+评论的免费下载模式,旨在节省大家寻找宝贵资源的时间,为大家带来珍贵的知识来源,但是从2021年7月27日起,因遭受注册轰炸和恶意爬虫造成网站负载过大导致宕机,被迫无奈,为减少网站运营时间成本,对部分资源进行捐赠下载,添加支付是反爬虫的最好方式,在此对未享受到免费下载福利的访客表示歉意;
(9)如遇支付页面未弹出、支付成功但未显示隐藏内容、资源链接失效或者提取码出错等情况请发送邮件到stubbornhuang@qq.com或者直接联系QQ:623256052,会尽快为您解决!加我时请备注原因,谢谢!或者点击按钮点击反馈进入反馈页面进行问题反馈
(10)由以上捐赠下载带来的不便请您谅解,如出现重复付费的情况,请联系我,我会尽快返回您的退款!
当前分类随机文章推荐
- Pytorch - nn.Transformer、nn.TransformerEncoderLayer、nn.TransformerEncoder、nn.TransformerDecoder、nn.TransformerDecoder参数详解 阅读2716次,点赞1次
- Python - list/numpy/pytorch tensor相互转换 阅读1710次,点赞0次
- Pytorch - .to()和.cuda()的区别 阅读826次,点赞0次
- Pytorch - pad_sequence、pack_padded_sequence、pack_sequence、pad_packed_sequence参数详解与使用 阅读957次,点赞0次
- Pytorch - 内置的CTC损失函数torch.nn.CTCLoss参数详解与使用示例 阅读1276次,点赞1次
- Pytorch - torch.nn.Conv2d参数详解与使用 阅读485次,点赞0次
- Pytorch - torch.stack参数详解与使用 阅读747次,点赞0次
- Pytorch - torch.optim优化器 阅读686次,点赞0次
- Pytorch - 为什么要设置随机数种子? 阅读548次,点赞0次
- Pytorch - transpose和permute函数的区别和用法 阅读1318次,点赞0次
全站随机文章推荐
- VPS - 使用XShell连接VPS 阅读3265次,点赞0次
- Mediapipe – 将Mediapipe HolisticTracking封装成动态链接库dll/so,实现在桌面应用中嵌入全身关节点识别、手势识别、抬手放手检测识别功能 阅读6550次,点赞15次
- Pytorch - .to()和.cuda()的区别 阅读826次,点赞0次
- 资源分享 - OpenGL Programming Guide (Seventh Edition) OpenGL红宝书英文第7版 英文高清PDF下载 阅读1691次,点赞0次
- Windows - OneDrive登录出现0x8019001错误 阅读361次,点赞0次
- WPF - 代码动态更换按钮的背景图片 阅读3240次,点赞0次
- 资源分享 - Introduction to 3D Game Programming with DirectX 12 英文高清PDF下载 阅读3253次,点赞1次
- 书籍翻译 – Fundamentals of Computer Graphics, Fourth Edition,第3章 Raster Images中文翻译 阅读2276次,点赞3次
- 资源分享 - Foundations of Game Engine Development, Volume 1 Mathematics 英文高清PDF下载 阅读3028次,点赞0次
- 资源分享 - Game Programming Golden Rules 英文高清PDF下载 阅读1557次,点赞0次
评论
169