Hello Frank,
Firstly thank you for the detailed explanation



 KFrank:

assuming that you’re performing “global” classification


Yes frank I’m performing global classification.



 KFrank:

are willing to flatten the input tensor


Will there be a need to do that?
my image input shapeat the starting is [batch_size,channels,image_size,image_size]
and by the time it comes to last layer, I believe the shape is [batch_size,64*8,4,4]
and I’m (want) converting this to shape  [batch_size,output_channels,1,1] after the convolution (then the linear and softmax stuff can come).
So now you know the shapes. what do u recommend i should do?