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
投稿猜你喜欢
jieba库是一款优秀的 Python 第三方中文分词库，jieba 支持三种分词模式：精确模式、全模式和搜索引擎模式，下面是三种模式的特点- 网页中使用flash可以增强页面的动态交互效果，特别是用flash来制作广告，效果更好。经常使用flash的人，可能就碰到了flash会遮住
- 摘要：Oracle和微软都是数据库方面的大厂商，采用两家的产品的企业也不少。今天这篇文章为大家对比Oracle和SQLServer的镜像。标
- 本文主要是介绍Go，从语言对比分析的角度切入。之所以选择与Python、Erlang对比，是因为做为高级语言，它们语言特性上有较大的相似性，
- theme: channing-cyan网页伪静态将 * 页伪装成静态网页，可以提升网页被搜索引擎检索道德概率表现形式为：网址看着像是一个具
- 之前整理发表了《XMLHTTPRequest的属性和方法简介》,它ajax要使用的核心的技术之一，现在就来实际运用它。这个Ajax标签导航，
- 表单递交合法性检测-只接受数字。如下代码加入HTML的<head>区：<SCRIPT LANGUAGE=&quo
- aspjpeg组件官方下载地址：http://www.persits.com/说明： 1、aspjpeg能对图片水印进行透明度调整
- 本文实例讲述了Python实现螺旋矩阵的填充算法。分享给大家供大家参考，具体如下：afanty的分析：关于矩阵（二维数组）填充问题自己动手推
- 我相信站长们做网站的最终目的还是想要获得收入的，我想象站长们大部分的都做Google的联盟的，我相信站长中大部分的人都有考虑过做英文站的，但
- 如下所示：两个函数：Basemap.drawparallels ##纬度 Basemap.drawmeridia
- 这个仿msn的右下角popup提示窗口效果很久以前收集的，现在整理出来给大家分享，需要的朋友可以拿去用，特点，提示窗口内容和js代码分离容易
- 目录[redis 调用Lua脚本](#redis 调用Lua脚本)[redis+lua 实现评分排行榜实时更新](#redis+lua 实现
- 一、高斯滤波 高斯滤波是一种线性平滑滤波，适用于消除高斯噪声，广泛应用于图像处理的减噪过程。 [1] 通俗的讲，高斯滤波就是对整幅图像进
- pyc的破解相对容易，使用cython将python文件编译成.so文件，能在一定程度上增强python源码的私密性。编译成.so文件环境准
- caller 属性返回一个对函数的引用，该函数调用了当前函数。functionName.caller functionName 对象是所执行
- NumPy数组的维数称为秩（rank），一维数组的秩为1，二维数组的秩为2，以此类推。在NumPy中，每一个线性的数组称为是一个轴（axes
- 基于Ubuntu16.04 + Python3 + nginx + mysql + Django接下来先安装这些必要的环境，这些操作都是在你
- 使用Django意味着后台框架的几乎所有内容都会和Django产生互动,排除功能全部手撸的情况.Django 后台admin有大量的属性和方
- 要读懂这些代码主要是要了解ASP中操作二进制数据的对象ADODB.Stream！本程序主要用的就是Adodb.Stream，如果你有这个基础
