Yes, the output should be an image, thats why i set the parameters in_channel and out_channel as 128, but I can’t understand why the expected input is basically inverted.
Edit: using (3,128,1) fixed the problem but should i change also out_channels to 3 if i need a 128x128 rgb image or is it correct to leave it as 128?
I’m not sure that I have correctly understood how conv2d works