Pergunta de entrevista da empresa TikTok

Write a cross-attention in PyTorch.