pytorch实现onehot编码转为普通label标签


Posted in Python onJanuary 02, 2020

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 multiprocessing.Manager介绍和实例(进程间共享数据)
Nov 21 Python
python 查找字符串是否存在实例详解
Jan 20 Python
浅谈Python 的枚举 Enum
Jun 12 Python
Python中getpass模块无回显输入源码解析
Jan 11 Python
TensorFlow实现Batch Normalization
Mar 08 Python
python pandas 如何替换某列的一个值
Jun 09 Python
mac下给python3安装requests库和scrapy库的实例
Jun 13 Python
selenium+PhantomJS爬取豆瓣读书
Aug 26 Python
python下载库的步骤方法
Oct 12 Python
使用keras实现孪生网络中的权值共享教程
Jun 11 Python
Python基于time模块表示时间常用方法
Jun 18 Python
Python 操作pdf pdfplumber读取PDF写入Exce
Aug 14 Python
pytorch标签转onehot形式实例
Jan 02 #Python
Python socket聊天脚本代码实例
Jan 02 #Python
解决Pytorch训练过程中loss不下降的问题
Jan 02 #Python
Pytorch evaluation每次运行结果不同的解决
Jan 02 #Python
Python for循环与getitem的关系详解
Jan 02 #Python
python实现双色球随机选号
Jan 01 #Python
PyTorch实现更新部分网络,其他不更新
Dec 31 #Python
You might like
linux使用crontab实现PHP执行计划定时任务
2014/05/10 PHP
浅谈php中fopen不能创建中文文件名文件的问题
2017/02/06 PHP
Yii 2中的load()和save()示例详解
2017/08/03 PHP
将string解析为json的几种方式小结
2010/11/11 Javascript
为JavaScript类型增加方法的实现代码(增加功能)
2011/12/29 Javascript
replace()方法查找字符使用示例
2013/10/28 Javascript
javascript实现点击后变换按钮显示文字的方法
2015/05/13 Javascript
JavaScript函数内部属性和函数方法实例详解
2016/03/17 Javascript
谈谈Vue.js——vue-resource全攻略
2017/01/16 Javascript
微信小程序开发的四十个技术窍门总结(推荐)
2017/01/23 Javascript
JavaScript获取ul中li个数的方法
2017/02/13 Javascript
微信小程序页面间传值与页面取值操作实例分析
2019/04/30 Javascript
vue2 中二级路由高亮问题及配置方法
2019/06/10 Javascript
如何基于原生javaScript生成带图片的二维码
2019/11/21 Javascript
webpack+vue.js构建前端工程化的详细教程
2020/05/10 Javascript
微信小程序自定义弹出层效果
2020/05/26 Javascript
javascript实现页面的实时时钟显示示例
2020/08/06 Javascript
vue实现前端列表多条件筛选
2020/10/26 Javascript
[12:51]71泪洒现场!是DOTA2让经典重现
2014/03/24 DOTA
[01:14]2014DOTA2展望TI 剑指西雅图newbee战队专访
2014/06/30 DOTA
理解Python中函数的参数
2015/04/27 Python
python中range()与xrange()用法分析
2016/09/21 Python
Python序列化基础知识(json/pickle)
2017/10/19 Python
python 删除指定时间间隔之前的文件实例
2018/04/24 Python
numpy中矩阵合并的实例
2018/06/15 Python
删除DataFrame中值全为NaN或者包含有NaN的列或行方法
2018/11/06 Python
基于wxPython的GUI实现输入对话框(2)
2019/02/27 Python
Python爬虫 urllib2的使用方法详解
2019/09/23 Python
Python使用Opencv实现图像特征检测与匹配的方法
2019/10/30 Python
购买原创艺术品:Zatista
2019/11/09 全球购物
汽车驾驶求职信
2013/10/25 职场文书
开业庆典答谢词
2014/01/18 职场文书
大学生毕业求职自荐书范文
2014/02/04 职场文书
cf战队收人广告词
2014/03/14 职场文书
鸟的天堂导游词
2015/01/31 职场文书
Python 居然可以在 Excel 中画画你知道吗
2022/02/15 Python