请按照下方说明实现 top_p_filter()。
probs 为 (V,) float32,8 <= V <= 1000,sum(probs) = 1,每个元素 > 0;0 < p <= 1。测试输入保证排序无并列。
随机分布 p=0.8
正确性调用
top_p_filter(tensor[10] [0.412188, 0.060504, 0.052582, 0.068733, 0.009947, 0.046436, …], 0.8)
预期结果
tensor[10] [0.502692, 0.073789, 0, 0.083825, 0, 0, …]
运行会检查可见示例,提交会运行完整的隐藏测试。
登录并提交后,你的提交记录会显示在这里。