Hi,
Could you add stack trace of error? The dimensions you have mentioned in your explanation do not exist in your code so I cannot find where it happens.
But I would like to mention that



 alex_gilabert:

(10, 512, 28 28) what would make it a matrix (10, 4014080) but i get an error message like this “shape ‘[10, 4014080]’ is invalid for input of size 4014080”.


512*28*28 = 401408 and 10*512*28*28 = 4014080 (note the zero at the end). So the error says that you cannot reshape 4014080 into [10, 4014080].
Bests