Cross Entropy Loss Function Python Code