请按照下方输入/输出格式实现 solve。
每个 head 的 value 维度相同。 将平均后的向量四舍五入到四位小数。
示例 1
输入
{"query":[1,0],"heads":[{"keys":[[1,0],[1,0]],"values":[[1,0],[0,1]]},{"keys":[[1,0],[1,0]],"values":[[2,0],[0,2]]}]}预期输出
[0.75,0.75]
示例 2
输入
{"query":[1,0],"heads":[{"keys":[[1,0],[1,0]],"values":[[4,0],[0,2]]},{"keys":[[1,0],[1,0]],"values":[[0,4],[2,0]]}]}预期输出
[1.5,1.5]
运行会检查可见示例，提交会检查完整测试集。
登录并提交后，你的提交记录会显示在这里。