我要招人登录

人工智能面试题更新 2026-08-05

手写Attention机制代码

华为HUAWEI人工智能电子/半导体编码实现技术原理PyTorch

考察说明

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

回答思路

能正确写出Q、K、V的点积注意力计算流程
正确使用缩放因子防止梯度消失
正确实现mask处理与softmax归一化
输出维度与输入维度匹配，体现对维度推导的掌握

本题已收录答题指导

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

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

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