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 无法相乘
版权声明：本文为weixin_44790486原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。