How To Create A Simple Neural Network Using Pytorch