请按照下方说明实现 beam_search()。
log_probs 为 float32 张量 (S, V),1 <= S <= 16,2 <= V <= 64;1 <= beam_width <= V;保证每步不存在分数并列。
S=4 V=5 beam=2
正确性调用
beam_search(tensor[4, 5] [-1.57314, -1.57694, -2.10813, -1.58211, -1.35095, -2.21237, …], 2)
预期结果
[[4, 3, 4, 2], -3.39176]
运行会检查可见示例,提交会运行完整的隐藏测试。
登录并提交后,你的提交记录会显示在这里。