Practice for NN use pytorch Use pytorch to make a simple neural network for handwritten number recognition