热心板栗

文章
5
资源
0
加入时间
3年0月21天

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 无法相乘...