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实现onehot编码转为普通label标签就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
本文实例讲述了Python数据结构之栈、队列及二叉树定义与用法。分享给大家供大家参考，具体如下： 目前只实现了三种，栈、队列和二叉树，哪天得空继续补吧~ 1. 栈 #栈 class...
假设我们有一个数据集，列名叫status下面有100万的数据，其中包装 “HUMAN_REFUSE”,”SYS_REFUSE”,”HUMAN_AGREE”,”SYS_APPROVING”...
注意：使用该方法，获取的数据总数目不能超过1万，否则出错 #在python3上运行 from elasticsearch import Elasticsearch from urll...
当用GOOGLE查的时候，内容几乎都是一样的。但是你想要的东西，一个也没有。例如，我就找不到中国人写的如何使用PYTHON来创建一个XML文件。当然，直接用文件写的方式也能够达到同样的效...
本文实例讲述了Python实现的KMeans聚类算法。分享给大家供大家参考，具体如下： 菜鸟一枚，编程初学者，最近想使用Python3实现几个简单的机器学习分析方法，记录一下自己的学习过...