pytorch中onehot编码转为普通label标签
摘要：
label转onehot的很多，但是onehot转label的有点难找，所以就只能自己实现以下，用的topk函数，不知道有没有更好的实现 one_hot = torch.tensor([[0,0,1],[0,1,0],[0,1,0]]) print(one_hot) label = torch.to 阅读全文
posted @ 2019-06-29 23:08 swuxyj 阅读(6649) 评论(3) 推荐(0) 编辑