拼多多集团-PDD面试题更新 2026-08-05

请写出Self-Attention的公式,并给出其代码实现。

拼多多集团-PDD人工智能电商编码实现技术原理PyTorch

考察说明

考察对Transformer核心注意力机制的数学原理与工程实现能力

回答思路

  1. 准确写出Q、K、V的线性变换及缩放点积注意力公式
  2. 解释缩放因子√d_k的作用
  3. 代码实现包含维度处理、softmax和mask处理
  4. 能说明复杂度O(n²d)及优化方向
本题已收录答题指导

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

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