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

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

使用Pytorch跑网络的时候发生了如标题所述的报错，好多地方有类似的情况，不同体现在最后是dimension或是dimension 1之类。

原因：下采样次数比上采样次数多一次，导致输出的图像尺寸比输入尺寸小一倍(此处网络里下采样一次长和宽减半，上采样则使长和宽增为原先的两倍)，最后在计算loss（MSE）时出现报错。只要将上采样次数增加或者将下采样次数减少，使两者次数相同，保持最后的输出尺寸与输入尺寸相同即可。