简单TransformerQKV

构建因果 Attention Mask

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

将每一行 Attention 分数转换为因果 softmax：位置 i 只能关注不大于 i 的位置，未来位置的概率必须为零。

分数矩阵为方阵，最多 64 行。将概率四舍五入到六位小数。

示例 1

输入

{"scores":[[1,0],[0,1]]}

预期输出

[[1.0,0.0],[0.268941,0.731059]]

示例 2

输入

{"scores":[[0,0,0],[0,0,0],[1,0,1]]}

预期输出

[[1.0,0.0,0.0],[0.5,0.5,0.0],[0.422319,0.155362,0.422319]]

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

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

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