I have my input image which is 128 x 96 size, and it is fed to the net, after a set of layers, I want to output the image of the same size.
My last layer now is ConvAct, and if to check the size of tensor, it is (10, 1, 96, 128), so to have a linear layer which would return me 128*96 image - what input size I should use?
(12288,…)?