中等其他

屏蔽 Padding Actions

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

将批量 action tensor 中的 padding 清零。给定形状为（B, T, D）的 padded actions 和长度向量 lens（B,），对样本 b 将所有 t ≥ lens[b] 的元素设为 0.0，合法位置保持不变。请使用 numpy。

每个 lens[b] 都在 1..T 范围内；actions 是（B, T, D）数组。

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

无 Padding

正确性

调用

mask_padded(tensor[2, 2, 2], float64 [1,2,3,4,5,6,7,8], tensor[2], int64 [2,2])

预期结果

[[[1, 2], [3, 4]], [[5, 6], [7, 8]]]

屏蔽第二步

正确性

调用

mask_padded(tensor[2, 2, 2], float64 [1,2,3,4,5,6,7,8], tensor[2], int64 [2,1])

预期结果

[[[1, 2], [3, 4]], [[5, 6], [0, 0]]]

Python 3 · NumPy · 函数题 · mask_padded()

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

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