Cross entropy loss python code. .