pytorch实现onehot编码转为普通label标签
更新时间：2020年01月02日 09:16:10 作者：swuxyj
今天小编就为大家分享一篇pytorch实现onehot编码转为普通label标签，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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 requests发送http请求的相关知识,requests是一个Python的第三方库,用于发送HTTP请求,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2023-12-12
这篇文章主要介绍了5行Python代码实现图像分割的步骤详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-05-05
在自动化工作中，有可能会遇到一些发送邮箱验证码类似的功能。本文将利用Python Selenium实现自动化读取网易邮箱验证码，感兴趣的可以了解一下2022-03-03
本文给大家分享的是使用python爬虫实现把《廖雪峰的 Python 教程》转换成PDF的方法和代码，有需要的小伙伴可以参考下2017-02-02
分析解决Python中sqlalchemy数据库连接池QueuePool异常
这篇文章主要来给大家分析sqlalchemy数据库连接池QueuePool的异常，给大家用详细的图文方式做出了解决的方案，有需要的朋友可以借鉴参考下，希望可以有所帮助2021-09-09
这篇文章主要介绍了django之自定义软删除Model的方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
在大多数场景中，我们都用 lxml 库解析网页源码，但你是否知道，lxml 库也是可以操作 svg 图片的。本文就来和大家聊聊具体操作方法，希望对大家有所帮助2023-01-01
这篇文章主要为大家详细介绍了python实现图书借阅系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-02-02
Python 中的变量不需要声明。每个变量在使用前都必须赋值，变量赋值以后该变量才会被创建。在 Python 中，变量就是变量，它没有类型，我们所说的"类型"是变量所指的内存中对象的类型。2017-02-02
相信大多数8090后都玩过太空射击游戏，在过去游戏不多的年代太空射击自然属于经典好玩的一款了，今天我们来自己动手实现它，在编写学习中回顾过往展望未来，下面开始讲解子弹与碰撞处理，在本课中，我们将添加玩家与敌人之间的碰撞，以及添加供玩家射击的子弹2022-08-08
最新评论