Pytorch Multiple Layer Perceptron