Domanda di colloquio di TikTok

Write a cross-attention in PyTorch.