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设计】。
如下所示： # -*- coding: utf-8 -*- # @Time :18-8-2 下午3:23 import sys reload(sys) sys.setdefault...
本文实例讲述了Python将多个映射合并为单个映射的方法。分享给大家供大家参考，具体如下： 问题：在逻辑上将多个字典或映射合并为一个单独的映射结构，以此执行某些特定的操作，比如查找值或者...
什么是迭代（iteration）呢？ 给定一个list或者tuple，通过for循环来遍历这个list或者tuple、这种遍历就是迭代（iteration）。只要是可迭代的对象都可以进行...
Python 中提供了对时间日期的多种多样的处理方式，主要是在 time 和 datetime 这两个模块里。今天稍微梳理一下这两个模块在使用上的一些区别和联系。 time 在 Pyth...
使用正则表达式的几个步骤： 1、用import re 导入正则表达式模块； 2、用re.compile()函数创建一个Regex对象； 3、用Regex对象的search()或finda...