unet 181line,should write as: if self.mode != 'ori': c2 = torch.cat([c2, self.ife1(c2)],dim=1) c3 = torch.cat([c3, self.ife2(c3)],dim=1)