中等TransformerAttention

实现缩放点积 Attention

请按照下方输入/输出格式实现 solve。

实现一个 query 的缩放点积 Attention。对 query-key 分数应用 softmax，并返回四舍五入到四位小数的加权 value 向量。

Keys 与 query 维度相同。输入和输出均为 JSON。

示例 1

输入

{"query":[1,0],"keys":[[1,0],[1,0]],"values":[[10,0],[0,20]]}

预期输出

[5.0,10.0]

示例 2

输入

{"query":[1,0],"keys":[[1,0],[0,1]],"values":[[10,0],[0,20]]}

预期输出

[6.6976,6.6048]

Python 3 · 标准输入 / 标准输出

运行会检查可见示例，提交会检查完整测试集。

登录并提交后，你的提交记录会显示在这里。