RuntimeError: Expected 3D (unbatched) or 4D (batched) input to conv2d, but got input of size: [1, 784]

[foti]

This diagram helped a lot understand and tackle this frequent issue.