Domanda di colloquio di TikTok

code cross entropy loss function.