TikTok Interview Question

code cross entropy loss function.