我要招人

后端岗位面试题更新 2026-08-05

请手写实现多头注意力机制（Multi-Head Attention）的核心代码。

字节跳动后端开发互联网/IT编码实现技术原理

考察说明

考察对注意力机制原理的理解和编码实现能力

回答思路

  1. 正确实现Q、K、V的线性变换和分头
  2. 准确计算缩放点积注意力及掩码处理
  3. 正确拼接多头输出并做输出变换
  4. 说明各步骤的维度变化和参数含义
本题已收录答题指导

本题附完整参考答案与评分标准

登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。

开始模拟面试登录查看答题指导