https://machinelearningmastery.com/building-multilayer-perceptron-models-in-pytorch/