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之字典,你还记得吗?
Sep 20 Python
python实现简单ftp客户端的方法
Jun 28 Python
深入解析Python设计模式编程中建造者模式的使用
Mar 02 Python
Python学习笔记之lambda表达式用法详解
Aug 08 Python
Python中IP地址处理IPy模块的方法
Aug 16 Python
django框架使用views.py的函数对表进行增删改查内容操作详解【models.py中表的创建、views.py中函数的使用,基于对象的跨表查询】
Dec 12 Python
PyTorch实现AlexNet示例
Jan 14 Python
使用TensorFlow直接获取处理MNIST数据方式
Feb 10 Python
Python内存映射文件读写方式
Apr 24 Python
python和node.js生成当前时间戳的示例
Sep 29 Python
Python 机器学习工具包SKlearn的安装与使用
May 14 Python
Django实现翻页的示例代码
May 24 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
php 高效率写法 推荐
2010/02/21 PHP
php学习笔记之 函数声明(二)
2011/06/09 PHP
[原创]ThinkPHP中SHOW_RUN_TIME不能正常显示运行时间的解决方法
2015/10/10 PHP
Laravel 5.3 学习笔记之 配置
2016/08/28 PHP
JQuery 表格操作(交替显示、拖动表格行、选择行等)
2009/07/29 Javascript
js 数组实现一个类似ruby的迭代器
2009/10/27 Javascript
判断浏览器的javascript版本的代码
2010/09/03 Javascript
从零开始学习jQuery (八) 插播:jQuery实施方案
2011/02/23 Javascript
javascript中setInterval的用法
2015/07/19 Javascript
Windows下用PyCharm和Visual Studio开始Python编程
2015/10/26 Javascript
JavaScript程序中的流程控制语句用法总结
2016/05/23 Javascript
javascript中数组(Array)对象和字符串(String)对象的常用方法总结
2016/12/15 Javascript
javascript过滤数组重复元素的实现方法
2017/05/03 Javascript
AngularJS基于provider实现全局变量的读取和赋值方法
2017/06/28 Javascript
JS实现的JSON数组去重算法示例
2018/04/11 Javascript
微信小程序实现的点击按钮 弹出底部上拉菜单功能示例
2018/12/20 Javascript
详解JavaScript修改注册表的方法
2020/01/05 Javascript
详解vue路由
2020/08/05 Javascript
浅析VUE防抖与节流
2020/11/24 Vue.js
[02:42]DOTA2英雄基础教程 杰奇洛
2013/12/23 DOTA
python使用nntp读取新闻组内容的方法
2015/05/08 Python
Python爬虫实战:分析《战狼2》豆瓣影评
2018/03/26 Python
pandas 选择某几列的方法
2018/07/03 Python
详解vscode实现远程linux服务器上Python开发
2020/11/10 Python
[原创]赚疯了!转手立赚800+?大佬的python「抢茅台脚本」使用教程
2021/01/12 Python
Marlies Dekkers内衣荷兰官方网店:荷兰奢侈内衣品牌
2020/03/27 全球购物
结构工程研究生求职信
2013/10/13 职场文书
汽车专业毕业生自荐信
2013/11/03 职场文书
最新的咖啡店创业计划书
2013/12/30 职场文书
学习保证书范文
2014/04/30 职场文书
党员示范岗材料
2014/12/19 职场文书
文案策划岗位职责
2015/02/11 职场文书
外出学习心得体会范文
2016/01/18 职场文书
资深HR教你写好简历中的自我评价
2019/05/07 职场文书
Oracle笔记
2021/04/05 Oracle
教你使用Ubuntu搭建DNS服务器
2022/09/23 Servers