RuntimeError: The size of tensor a (128) must match the size of tensor b (256) at non-singleton dimension 3

这个错误通常是因为你在进行某些操作时，两个张量的维度不匹配导致的。在这个具体的错误中，你正在对两个不同大小的张量进行操作，一个张量的第四个维度的大小为128，而另一个张量的第四个维度的大小为256，这导致了错误。你需要确保两个张量在进行操作时，它们的维度匹配。你可以通过使用reshape或者transpose等操作来改变张量的维度来匹配它们的大小。