Cross Entropy Loss Function In Python