pytorch实现onehot编码转为普通label标签
作者：swuxyj 发布时间：2023-04-18 10:19:21
标签：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])
来源：https://www.cnblogs.com/youmuchen/p/11108247.html
0
投稿猜你喜欢
- 当我们准备建立一个Web站点时，就必须向域名登记机构申请一个Internet域名，因此，我们通常希望了解自己准备使用的域名是否已经被注册，这
1.介绍PDF 格式是与平台无关，它独立于底层操作系统和渲染引擎。事实上，PDF 是基于一种脚本语言—&mdas- 将json转为结构体时，经常会遇到无法确定某个字段类型的情况。在Go中可以使用interface 任意类型来解决。// convert js
- 一、绘制总体图形import numpy as npimport matplotlib.pyplot as pltfrom mpl_tool
- 一、图像缩略图的编辑图像的缩略图是指把图像按原比例缩小,可作为原图的预览，这在网络速度比较慢时可快速地显示图片的概图。当你的网页上有大型图片
- 测试的目录如下：root ├── module_root.py ├── package_a&
- 安装了python3之后，会有pip31. 使用pip install XXX新安装的库会放在这个目录下面python2.7/site-pa
- 前言jieba 基于Python的中文分词工具，安装使用非常方便，直接pip即可，2/3都可以，功能强悍，十分推荐。中文分词(Chinese
- 作用: 构建一些简单的SQL语句，结合在提交表单时使用，可以较方便<％@LANGUAGE="VBSCRIPT&
- 就在刚才，我给GMAIL安装了Twitter插件。安装完插件后，Gmail提示我要连接Twitter的账号才能使用：我顺着提示操作下去，就看
- 本文实例讲述了Python使用xlrd读取Excel格式文件的方法。分享给大家供大家参考。具体如下：使用xlrd能够很方便的读取excel文
- 为了建设班级主页，买了个空间，支持SA FileUp组件。鼓弄了一天，终于有了大致的了解，下面是我的实例，希望对大家有所帮助。大家可以根据自
- 最近我在用梯度下降算法绘制神经网络的数据时，遇到了一些算法性能的问题。梯度下降算法的代码如下（伪代码）：def gradient_desce
- 代数运算定义加法运算图像叠加需要满足该条件：两幅图像的像素必须相同（尺寸，形状）减法运算乘法运算1、图像的局部显示。2、用二值蒙版图像与原图
- 我的数据库和报表服务的版本如下：数据库：SQL Server 2008 R2报表服务：SQL Server 2008 R2 Reportin
- typeof都返回object在JavaScript中所有数据类型严格意义上都是对象，但实际使用中我们还是有类型之分，如果要判断一个变量是数
- 1 The syntax of the SQL statement is verified.SQL的语法检查2 The data dicti
- 本文实例为大家分享了vuex实现购物车功能的具体代码，供大家参考，具体内容如下页面布局：采用了element-ui的表格对商品列表和购物车列
- 项目需要就在现有的服务器上面重新安装了个mysql服务器，还挺费劲儿呢，因为之前都是在我的笔记本上面试验的，它的系统是Ubuntu的，什么路
- 版本一：按分隔符进行比较 算法思路：按分隔符截取搜索字符串循环与待搜索字符进行比较 USE [Fly] GO --参数：@inStr 待搜索
