Cross Entropy Loss Function Python Programming