This is a basic implementation of a neural network using pytorch. The network accepts an input of 784 and generate a final output of 10. eg(classifying 28x28 -> 784 images of 10 digits)