Cross Entropy Loss Function Python Definition