How to get indices of top-K values from a numpy array

How to get indices of top-K values from a numpy array

np.argpartition(probs,-5)[-5:]

posted @ 2023-03-30 16:24  不要肥宅  阅读(41)  评论(0)  编辑  收藏  举报