Implement `categorical_cross_entropy`. Libraries (NumPy, PyTorch, SciPy) are allowed, but Pure Python implementations earn bonus XP!
categorical_cross_entropy(*args, **kwargs)