2021年1月12日
pytorch one hot 的转换
摘要： 转onehot one_hot = F.one_hot(label.long(), num_classes=n_classes) 转回来 label = torch.argmax(one_hot, -1) 阅读全文
posted @ 2021-01-12 10:20 mrbean 阅读(845) 评论(0) 推荐(0) 编辑