RuntimeError: mat1 and mat2 shapes cannot be multiplied (60000x150 and 784x10)

Question

For a school homework I am writing a little pytorch code, that should learn to decide from 10 different shapes, what type of shape the input is. The following code shows, ho