qq_45013475的博客

原创 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）时出现报错，只要将上采样次数增加，使得最后输出尺寸