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之wxPython菜单使用详解
Sep 28 Python
python开发之thread实现布朗运动的方法
Nov 11 Python
django.db.utils.ProgrammingError: (1146, u“Table‘’ doesn’t exist”)问题的解决
Jul 13 Python
python导入模块交叉引用的方法
Jan 19 Python
python 输出列表元素实例(以空格/逗号为分隔符)
Dec 25 Python
Pytorch技巧:DataLoader的collate_fn参数使用详解
Jan 08 Python
Python求凸包及多边形面积教程
Apr 12 Python
Python爬虫获取页面所有URL链接过程详解
Jun 04 Python
keras实现VGG16 CIFAR10数据集方式
Jul 07 Python
聊聊python中的异常嵌套
Sep 01 Python
Python基于template实现字符串替换
Nov 27 Python
python 实现socket服务端并发的四种方式
Dec 14 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编写PDF文档生成器
2006/10/09 PHP
php中常用字符串处理代码片段整理
2011/11/07 PHP
PHP中实现中文字符进制转换原理分析
2011/12/06 PHP
PHP优于Node.js的五大理由分享
2012/09/15 PHP
destoon数据库表说明汇总
2014/07/15 PHP
php的api数据接口书写实例(推荐)
2016/09/22 PHP
PHP中关于php.ini参数优化详解
2020/02/28 PHP
PNGHandler-借助JS让PNG图在IE下实现透明(包括背景图)
2007/08/31 Javascript
De facto standard 世界上不可思议的事实标准
2010/08/29 Javascript
jQuery)扩展jQuery系列之一 模拟alert,confirm(一)
2010/12/04 Javascript
javascript 系统文件夹文件操作及参数介绍
2013/01/08 Javascript
js带按钮的提示框可供选择示例代码
2013/09/17 Javascript
JavaScript学习笔记之JS函数
2015/01/22 Javascript
JS 调用微信扫一扫功能
2016/12/22 Javascript
Bootstrap多级菜单的实现代码
2017/05/23 Javascript
页面缩放兼容性处理方法(zoom,Firefox火狐浏览器)
2017/08/29 Javascript
VUE饿了么树形控件添加增删改功能的示例代码
2017/10/17 Javascript
layui之select的option叠加问题的解决方法
2018/03/08 Javascript
vue.js中导出Excel表格的案例分析
2019/06/11 Javascript
[01:46]2020完美世界全国高校联赛秋季赛报名开启
2020/10/15 DOTA
python getopt详解及简单实例
2016/12/30 Python
python学习之matplotlib绘制散点图实例
2017/12/09 Python
python中的for循环
2018/09/28 Python
Python 内存管理机制全面分析
2021/01/16 Python
阿里健康大药房:阿里自营网上药店
2017/08/01 全球购物
洛佩桑酒店官方网站:Lopesan Hotels
2019/04/15 全球购物
Prototype中如何为一个元素添加一个方法
2014/12/08 面试题
电脑专业个人求职信范文
2014/02/04 职场文书
护士自我鉴定怎么写
2014/02/07 职场文书
一年级班主任感言
2014/03/08 职场文书
公司年终奖分配方案
2014/06/16 职场文书
家长给老师的感谢信
2015/01/20 职场文书
党员带头倡议书
2015/04/29 职场文书
清明扫墓感想
2015/08/11 职场文书
大学生干部培训心得体会
2016/01/06 职场文书
flex布局中使用flex-wrap实现换行的项目实践
2022/06/21 HTML / CSS