Mismatched output size when using convolution layer [Figure] - PyTorch ...

Responsive image
source