Pregunta de entrevista de TikTok

Write a cross-attention in PyTorch.