Question d’entretien chez TikTok

Write a cross-attention in PyTorch.