简单Transformer

构建 Causal Attention Mask

请按照下方说明实现 causal_mask()。

构建长度为 n 的 causal（因果）Attention Mask。返回 n × n 矩阵：当位置 j 对位置 i 可见（j ≤ i）时，元素 (i, j) 为 1，否则为 0。这样可以阻止 token 关注未来 token。

n 是 1 到 32 的整数。

检查项正确性形状数据类型

长度 3

正确性

调用

causal_mask(3)

预期结果

[[1, 0, 0], [1, 1, 0], [1, 1, 1]]

长度 1

边界情况

调用

causal_mask(1)

预期结果

[[1]]

Python 3 · Python · 函数题 · causal_mask()

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

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