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根据unicode判断语言类型，具体如下。 实例代码： def is_chinese(uchar): """判断一个unicode是否是汉字"""...
什么是状态机？ 关于状态机的一个极度确切的描述是它是一个有向图形，由一组节点和一组相应的转移函数组成。状态机通过响应一系列事件而“运行”。每个事件都在属于“当前”节点的转移函数的控制范围...
如下所示： list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] slice = random.sample(list, 5) #从list中随机获取5个元...
本文实例为大家分享了Python OpenCV图像直方图和反向投影的具体代码，供大家参考，具体内容如下 当我们想比较两张图片相似度的时候，可以使用这一节提到的技术 直方图对比 反向投影...
Python 私有函数的实例详解 与大多数语言一样，Python 也有私有的概念： • 私有函数不可以从它们的模块外面被调用 • 私有类方法不能够从它们的类外面...