albanD:

model2_out = self.model2(x) model3_out = self.model3(x) z = torch.cat([model2_out, model3_out], 1)


Thanks a lot  
I passed the identity layers in the init functions and in the forward I accepted your suggestion .