Interpolate expects the 2nd dimension to be “channel” dimension. I’m not sure what your [32, 68] tensor is supposed to represent, but if you want to be able to add it back to the result, interpolate would expect the size argument to have a single value, like  F.interpolate(residual1, size=[1024])