Top K算子 #23
Description
topK(Tensor indices, Tensor probs, Tensor logits, int64 k)
根据logits最后一维选出最大的k个值（若k为负数则选最小的k个值），值放在probs里，序号放在indices里。
- CPU
- CUDA
- BANG
{{ message }}
topK(Tensor indices, Tensor probs, Tensor logits, int64 k)
根据logits最后一维选出最大的k个值（若k为负数则选最小的k个值），值放在probs里，序号放在indices里。