字节跳动面试题更新 2026-08-05

如果用 TP=2 的策略,self-attention 层里的 QKV 矩阵具体是怎么切分的?

字节跳动人工智能互联网/IT系统设计技术原理

考察说明

考察张量并行下 QKV 线性层的切分方式与通信开销

回答思路

  1. 明确 Q、K、V 权重按列切分、输出层按行切分
  2. 指出 TP 通信在 forward 与 backward 中的位置
  3. 说明切分与头数、隐藏维度的关系,以及兼容性约束
本题已收录答题指导

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

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