网易游戏雷火面试题更新 2026-08-05

请用PyTorch编写一个Transformer的多头注意力模块,并说明其余结构的实现思路。

网易游戏雷火人工智能编码实现问题拆解技术原理PyTorch

考察说明

考察Transformer多头注意力的编码实现能力与整体架构理解

回答思路

  1. 正确实现Q、K、V线性投影与分头
  2. 实现缩放点积注意力与mask处理
  3. 正确拼接多头输出并做输出投影
  4. 能清晰说明编码器、解码器、位置编码等其余结构
  5. 代码风格清晰,能处理张量维度
本题已收录答题指导

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

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