Python Cross Entropy Loss Formula