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  不要肥宅  阅读(63)  评论(0)    收藏  举报