torch.argsort(output, 1)[0][-num_candidates:]
时间: 2023-03-14 16:08:02 浏览: 130
torch-1.7.1+cu110-cp37-cp37m-linux_x86_64.whl离线安装包linux系统x86_64
这里的 num_candidates 指定了返回的最大候选结果数量,torch.argsort(output, 1)[0][-num_candidates:] 就是根据给定的输出(output),对其进行排序,并返回最后 num_candidates 个结果。
阅读全文