获取tensor的非重复数字

np.unique(labels.cpu().detach().numpy())

posted @ 2022-03-15 09:34  小kk_p  阅读(112)  评论(0)    收藏  举报