part 1: create feedforward network, print output
pytorch network1: Create simple feedforward network, print the output
part 2: print prediciton, loss, create optimizer, calculate gradients, and run optimizer
pytorch network2: print prediction, loss, run backprop, run training optimizer
(Edit: note, this hsould be log_softmax, rather than softmax  )