Apologies for the mistake. It’s not reshape… please do a torch.cat as shown in this link

  
    
    
    RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[1, 8, 600, 800] to have 3 channels, but got 8 channels instead vision
  
  
    how can I do this in pytorch?