转onehot
one_hot = F.one_hot(label.long(), num_classes=n_classes)
转回来
label = torch.argmax(one_hot, -1)
相关文章：
猜你喜欢
相关资源
相似解决方案
热门标签
转onehot
one_hot = F.one_hot(label.long(), num_classes=n_classes)
转回来
label = torch.argmax(one_hot, -1)
相关文章：