topK(Tensor indices, Tensor probs, Tensor logits, int64 k) 根据logits最后一维选出最大的k个值（若k为负数则选最小的k个值），值放在probs里，序号放在indices里。 - [ ] CPU - [ ] CUDA - [ ] BANG
topK(Tensor indices, Tensor probs, Tensor logits, int64 k)
根据logits最后一维选出最大的k个值（若k为负数则选最小的k个值），值放在probs里，序号放在indices里。