获取tensor的非重复数字

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

posted @ 2022-03-15 09:34  小丑_jk  阅读(102)  评论(0)  编辑  收藏  举报