Solved by ptrblck in post #4 
              
              
                In your code snippet, 128*7*7 is interpreted as 6272. 
The way to write the input shape as a multiplication makes instead of the result directly makes it easier to understand the shape. 
E.g. when this linear layer is preceded by conv layers, you would flatten the conv activation, so that it’ll have…