RuntimeError: Expected 4-dimensional input for 4-dimensional weight [64, 3, 7, 7], but got 3-dimensional input of size [8, 600, 800] instead

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