1. 首页
  2. 文章中心
  3. 代码问题

RuntimeError: batch1 dim 2 must match batch2 dim 1

80 阅读 0 评论
我是靠谱客的博主 热心板栗，最近开发中收集的这篇文章主要介绍RuntimeError: batch1 dim 2 must match batch2 dim 1，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

RuntimeError: batch1 dim 2 must match batch2 dim 1
这个问题就是两个特征的维度要匹配
比如，这个代码
A = torch.bmm(exemplar_corr, query_flat)
exemplar_corr 特征大小为[96, 49, 1024])
query_flat 特征大小为 [96, 2048, 49]
那这样的话，就会报上面的错误，1024 和2048 无法相乘

最后

以上就是热心板栗为你收集整理的RuntimeError: batch1 dim 2 must match batch2 dim 1的全部内容，希望文章能够帮你解决RuntimeError: batch1 dim 2 must match batch2 dim 1所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
点赞(53)

相关文章

【年终总结】2019年有三AI做了什么，2020年我们要做什么？
【年终总结】2019年有三AI做了什么，2020年我们要做什么？
RuntimeError mat1 dim 1 must match mat2 dim 0RuntimeError: mat1 dim 1 must match mat2 dim 0
RuntimeError mat1 dim 1 must match mat2 dim 0RuntimeError: mat1 dim 1 must match mat2 dim 0
PyTorch错误 torch.addmm RuntimeError: mat1 dim 1 must match mat2 dim 0
PyTorch错误 torch.addmm RuntimeError: mat1 dim 1 must match mat2 dim 0
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法解决方法：
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法解决方法：
RuntimeError: batch1 dim 2 must match batch2 dim 1
RuntimeError: batch1 dim 2 must match batch2 dim 1
RuntimeError: mat1 dim 1 must match mat2 dim 0
RuntimeError: mat1 dim 1 must match mat2 dim 0
RuntimeError: mat1 dim 1 must match mat2 dim 0以及local variable ‘beta1‘ referenced before assignment报错1：
RuntimeError: mat1 dim 1 must match mat2 dim 0以及local variable ‘beta1‘ referenced before assignment报错1：
这便是有三AI一年的底蕴，那些5000粉丝1000阅读量的AI技术干货
这便是有三AI一年的底蕴，那些5000粉丝1000阅读量的AI技术干货

评论列表共有 0 条评论

立即
投稿
返回
顶部