Pergunta de entrevista da empresa TikTok

Write self-attention and cross-entropy pseudocode