页面加载中…
AI006. 因果注意力计算
困难通过率 100% · 提交 1 · 通过 1
华为 AI 岗AttentionSoftmax矩阵
本题改编自算法机考高频题型 · 考点：华为 AI 岗 / Attention · 难度：困难
给定同一序列的 Q、K、V 三个矩阵，请计算单头因果自注意力输出。第 i 个 token 只能关注编号不大于 i 的 token。注意力分数为 dot(Q[i], K[j]) / sqrt(d)，权重为这些分数的 Softmax，输出为对应 V 的加权和。
这类题属于算法机考高频题型中「华为 AI 岗 / Attention」方向的高频题型，通常考察对「华为 AI 岗 / Attention」的建模能力与边界条件处理。掌握本题的解题思路后，可举一反三应对同类真题方向，稳步提升机考通过率。
输入描述
第一行输入 n d。接下来 n 行输入 Q 矩阵，再接下来 n 行输入 K 矩阵，最后 n 行输入 V 矩阵。每行 d 个实数。
输出描述
输出 n 行，每行 d 个实数，表示注意力输出。每个数保留四位小数，绝对值小于 0.00005 时输出 0.0000。
示例
示例 1
输入示例
2 1 0 0 0 0 2 4
输出示例
2.0000 3.0000
说明：
一维两 token 的平均效果
示例 2
输入示例
3 1 1 1 1 0 0 10 1 2 100
输出示例
1.0000 1.5000 99.9911
说明：
后面的 token 不得影响前面输出
时间限制 3000 ms · 内存限制 256 MB
本平台为独立第三方培训机构，与华为技术有限公司无任何关联；课程的服务内容与权益以购买协议为准，学习效果因个人情况而异。「华为 OD」「华为可信」等仅为对岗位与考试方向的客观描述，相关商标归各自权利人所有。
本题暂无题解精讲。
我的提交
登录后可查看你在本题的历史提交，以及每次的各用例通过情况。
