互联网/IT行业面试题更新 2026-08-05

请用代码实现一个基础的自注意力(Self-Attention)机制。

快手人工智能互联网/IT编码实现技术原理PyTorch

考察说明

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

回答思路

  1. 正确实现Q、K、V的线性变换
  2. 正确计算缩放点积注意力并添加mask(可选)
  3. 正确进行softmax归一化与输出投影
  4. 代码简洁、可读,能解释各步骤含义
本题已收录答题指导

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

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