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设计】。
前言 VScode是一个相当优秀的IDE，具备开源、跨平台、模块化、插件丰富、启动时间快、颜值高、可高度定制等等优秀的特质，不愧是微软爸爸的私生子。 所以用VScode来编写Pyth...
本文实例讲述了Python实现比较两个文件夹中代码变化的方法。分享给大家供大家参考。具体如下： 这里将修改代码后的目录与原始目录做对比，罗列出新增的代码文件，以及修改过的代码文件 #...
最近在学习 python 语言。大致学习了 python 的基础语法。觉得 python 在数据处理中的地位和它的 list 操作密不可分。 特学习了相关的基础操作并在这里做下笔记。...
首先,需要导入random模块: import random 随机取1-33之间的1个随机数，可能重复: random.choice(range(1,34)) print得到...
最近有个需求，是做两个数组重复程度计算，麻烦就麻烦在单个数组的元素有可能重复，处理思路如下： 1. 找到重复元素 2. 元素个数统计，利用np.bincount转换，即元素个数统计到元素...