RuntimeError: Expected 4-dimensional input for 4-dimensional weight [12544, 1, 3, 3], but got 2-dimensional input of size [128, 12544] instead

Question

I\'m trying to use Convolutional Auto-Encoder Decoder. My input image size : 112*112 Batch : 128 iuput Channel = 1 (grayscale)

and Error occur when the input goes into fi