哔哩哔哩面试题更新 2026-08-05

手撕多头注意力机制的实现,并解释为什么缩放因子是根号下dk。

哔哩哔哩人工智能编码实现技术原理PyTorch

考察说明

考察多头注意力机制的实现细节与缩放因子的数学原理

回答思路

  1. 能写出Q、K、V线性变换与头部切分
  2. 能写出注意力分数计算与softmax归一化
  3. 能解释缩放因子防止点积过大导致梯度消失
  4. 能说明多头拼接与输出投影
本题已收录答题指导

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

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