The shape mismatch error is raised in the torch.cat operation since x and skip have different spatial sizes. Pad one of these tensors or slice the other one to create matching spatial sizes.
The shape mismatch error is raised in the torch.cat operation since x and skip have different spatial sizes. Pad one of these tensors or slice the other one to create matching spatial sizes.