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
最新推荐文章于 2023-02-10 16:17:03 发布
本文详细解析了RuntimeError:batch1dim2必须匹配batch2dim1的出现原因,以PyTorch为例,通过实际代码中的维度调整来演示如何避免这一常见错误。重点在于确保特征矩阵相乘时的维度一致性。
摘要由CSDN通过智能技术生成