The size of tensor a (64) must match the size of tensor b (3) at non-singleton dimension 1

这个错误的意思是，您正在尝试执行一个需要a和b张量在第1个维度上具有相同大小的操作，但是它们的大小不同。例如，a张量可能具有形状（2, 64, 3），而b张量可能具有形状（2, 3, 3）。在这种情况下，您需要重新调整其中一个张量的大小，使它们在第1个维度具有相同的大小。您可以使用PyTorch的resize()或reshape()函数来完成这项任务。