Hello, trying to train model on 32x32 images. but  got dimensional error.
Here is my Module.

image1366×768 115 KB

And here is an error

image1366×768 138 KB

i just don’t get where is that 4th input?