one thing i want to know is, in my understanding we are flattening the features in last convolution then the neural net if for
400->120
120->84
84->10
if that’s correct, why don’t we just make the neural net for
32*32->120
120->84
84->10