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
最新推荐文章于 2024-02-07 04:35:38 发布