这篇文章给大家分享的是有关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.topk(one_hot, 1)[1].squeeze(1) print(label)
tensor([[0, 0, 1], [0, 1, 0], [0, 1, 0]]) tensor([2, 1, 1])
pytorch的优点
1.PyTorch是相当简洁且高效快速的框架；2.设计追求最少的封装；3.设计符合人类思维，它让用户尽可能地专注于实现自己的想法；4.与google的Tensorflow类似，FAIR的支持足以确保PyTorch获得持续的开发更新；5.PyTorch作者亲自维护的论坛 供用户交流和求教问题6.入门简单
感谢各位的阅读！关于“pytorch如何实现onehot编码转为普通label标签”这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，让大家可以学到更多知识，如果觉得文章不错，可以把它分享出去让更多的人看到吧！
![[Pytorch框架] 5.1 kaggle介绍](https://www.kunjuke.com/file/css/thumb/9.jpg/280-180.jpg)
