pytorch标签转onehot形式实例


Posted in Python onJanuary 02, 2020

代码:

import torch

class_num = 10
batch_size = 4
label = torch.LongTensor(batch_size, 1).random_() % class_num
print(label.size())

one_hot = torch.zeros(batch_size, class_num).scatter_(1, label, 1)
print(one_hot)

输出:

torch.Size([4, 1])
tensor([[0., 0., 0., 0., 0., 0., 0., 1., 0., 0.],
[0., 0., 0., 0., 0., 1., 0., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 1., 0., 0., 0.],
[0., 0., 0., 0., 0., 0., 0., 0., 0., 1.]])

注意:

label的形状必须是[n,1]的,也就是必须是二维的,且第二个维度长度为1,如果是一维度的,则需要升维度,代码如下:

import torch

class_num = 10
batch_size = 4
label = torch.LongTensor(batch_size).random_() % class_num
print(label.size())
label = torch.unsqueeze(label,dim=1)
print(label.size())

以上这篇pytorch标签转onehot形式实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python学生信息管理系统修改版
Mar 13 Python
python使用生成器实现可迭代对象
Mar 20 Python
python docx 中文字体设置的操作方法
May 08 Python
浅谈python str.format与制表符\t关于中文对齐的细节问题
Jan 14 Python
python与字符编码问题
May 24 Python
Django框架中间件(Middleware)用法实例分析
May 24 Python
浅谈PySpark SQL 相关知识介绍
Jun 14 Python
Python 忽略文件名编码的方法
Aug 01 Python
Python 通过爬虫实现GitHub网页的模拟登录的示例代码
Aug 17 Python
python制作微博图片爬取工具
Jan 16 Python
MATLAB 如何求取离散点的曲率最大值
Apr 16 Python
python批量更改目录名/文件名的方法
Apr 18 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
pytorch中的transforms模块实例详解
Dec 31 #Python
You might like
php学习笔记之 函数声明
2011/06/09 PHP
PHP中空字符串介绍0、null、empty和false之间的关系
2012/09/25 PHP
PHP使用CURL实现对带有验证码的网站进行模拟登录的方法
2014/07/23 PHP
PHP编程计算两个时间段是否有交集的实现方法(不算边界重叠)
2017/05/30 PHP
实例讲解PHP页面静态化
2018/02/05 PHP
use jscript List Installed Software
2007/06/11 Javascript
Exjs 入门篇
2010/04/07 Javascript
DIV菜单层实现代码
2010/11/19 Javascript
Javascript公共脚本库系列(一): 弹出层脚本
2011/02/24 Javascript
js 实现图片预加载(js操作 Image对象属性complete ,事件onload 异步加载图片)
2011/03/25 Javascript
使用typeof判断function是否存在于上下文
2014/08/14 Javascript
JavaScript跨浏览器获取页面中相同class节点的方法
2015/03/03 Javascript
jquery实现翻动fadeIn显示的方法
2015/03/05 Javascript
jquery京东商城双11焦点图多图广告特效代码分享
2015/09/06 Javascript
jQuery自定义滚动条完整实例
2016/01/08 Javascript
Vue.js创建Calendar日历效果
2016/11/03 Javascript
从零开始封装自己的自定义Vue组件
2018/10/09 Javascript
React事件处理的机制及原理
2018/12/03 Javascript
说说如何在Vue.js中实现数字输入组件的方法
2019/01/08 Javascript
JS实现json数组排序操作实例分析
2019/10/28 Javascript
[01:16:01]VGJ.S vs Mski Supermajor小组赛C组 BO3 第一场 6.3
2018/06/04 DOTA
[36:22]VP vs Serenity 2018国际邀请赛小组赛BO2 第一场 8.16
2018/08/17 DOTA
[01:02:32]DOTA2-DPC中国联赛 正赛 iG vs PSG.LGD BO3 第二场 2月26日
2021/03/11 DOTA
Python输出带颜色的字符串实例
2017/10/10 Python
python实现监控某个服务 服务崩溃即发送邮件报告
2018/06/21 Python
解决Django中修改js css文件但浏览器无法及时与之改变的问题
2019/08/31 Python
Django ValuesQuerySet转json方式
2020/03/16 Python
浅谈python3打包与拆包在函数的应用详解
2020/05/02 Python
CSS3实现大小不一的粒子旋转加载动画
2016/04/21 HTML / CSS
Philosophy美国官网:美国美容品牌
2016/08/15 全球购物
欧舒丹俄罗斯官方网站:L’OCCITANE俄罗斯
2019/11/22 全球购物
会计专业的自荐信
2013/12/12 职场文书
网上卖盒饭创业计划书范文
2014/02/07 职场文书
省级优秀毕业生主要事迹
2014/05/29 职场文书
关于诚信的活动方案
2014/08/18 职场文书
护士长2014年度工作总结
2014/11/11 职场文书