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设计】。
先看看效果： 在linux的终端中，ANSI转义序列来控制颜色 基本规则: 前面加上\033[,结尾用\033[0m重置为原来的颜色 可以在终端中输入下面这句，就可以看到输出绿色的h...
Python数据结构与算法（几种排序） 数据结构与算法（Python） 冒泡排序 冒泡排序（英语：Bubble Sort）是一种简单的排序算法。它重复地遍历要排序的数列，一次比较两个...
列表推导式 你有一个list： bag = [1, 2, 3, 4, 5] 现在你想让所有元素翻倍，让它看起来是这个样子： [2, 4, 6, 8, 10] 大多初学者，根据之前语言的经...
python生成二维码的实例详解 版本相关 操作系统：Mac OS X EI Caption Python版本：2.7 IDE：Sublime Text 3 依赖库 Pyth...
Python产生一个数值范围内的不重复的随机数，可以使用random模块中的random.sample函数，其用法如下： import random random.sample(po...