Machine Learning Loss Function Pytorch Tutorial Example