Machine Learning Loss Function Pytorch Tutorial Mlp