Machine Learning Loss Function Pytorch Tutorial