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 相关文章推荐
Android应用开发中Action bar编写的入门教程
Feb 26 Python
python中实现数组和列表读取一列的方法
Apr 03 Python
Python 十六进制整数与ASCii编码字符串相互转换方法
Jul 09 Python
Python多继承原理与用法示例
Aug 23 Python
如何在Django中添加没有微秒的 DateTimeField 属性详解
Jan 30 Python
python日志模块logbook使用方法
Sep 19 Python
python自动化实现登录获取图片验证码功能
Nov 20 Python
Pytorch Tensor 输出为txt和mat格式方式
Jan 03 Python
python设置环境变量的作用整理
Feb 17 Python
解决Python 函数声明先后顺序出现的问题
Sep 02 Python
pandas使用函数批量处理数据(map、apply、applymap)
Nov 27 Python
python 镜像环境搭建总结
Sep 23 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教程 插件机制在PHP中实现方案
2012/11/02 PHP
Laravel4中的Validator验证扩展用法详解
2016/07/26 PHP
php头像上传预览实例代码
2017/05/02 PHP
Yii 2.0在Grid中格式化时间方法示例
2017/06/06 PHP
laravel框架中路由设置,路由参数和路由命名实例分析
2019/11/23 PHP
PHP检查文件是否存在,不存在自动创建及读取文件内容操作示例
2020/01/23 PHP
正则表达式判断是否存在中文和全角字符和判断包含中文字符串长度
2008/09/27 Javascript
js编码之encodeURIComponent使用介绍(asp,php)
2012/03/01 Javascript
Javascript加载速度慢的解决方案
2014/03/11 Javascript
jQuery遍历Table应用示例
2014/04/09 Javascript
jquery map方法使用示例
2014/04/23 Javascript
详谈javascript中的cookie
2015/06/03 Javascript
在JavaScript的正则表达式中使用exec()方法
2015/06/16 Javascript
AngularJS表单基本操作
2017/01/09 Javascript
JS实现的合并多个数组去重算法示例
2018/04/11 Javascript
ExtJs使用自定义插件动态保存表头配置(隐藏或显示)
2018/09/25 Javascript
Vue利用History记录上一页面的数据方法实例
2018/11/02 Javascript
es6中reduce的基本使用方法
2019/09/10 Javascript
[01:15:44]首部DOTA2纪录片今日23时全网上映
2014/03/19 DOTA
Python3 socket同步通信简单示例
2017/06/07 Python
python实战教程之自动扫雷
2018/07/13 Python
opencv python 图像轮廓/检测轮廓/绘制轮廓的方法
2019/07/03 Python
python中常见错误及解决方法
2020/06/21 Python
Python3中FuzzyWuzzy库实例用法
2020/11/18 Python
HTML5对比HTML4的主要改变和改进总结
2016/05/27 HTML / CSS
Sahajan美国:阿育吠陀护肤品牌
2021/01/09 全球购物
Linux中如何用命令创建目录
2015/01/12 面试题
北大青鸟学生求职信
2013/09/24 职场文书
航空大学应届生求职信
2013/11/10 职场文书
资源环境与城乡规划管理专业自荐书
2014/09/26 职场文书
2014年社区党建工作汇报材料
2014/11/02 职场文书
2014年科室工作总结
2014/11/20 职场文书
公路施工安全责任书
2015/05/08 职场文书
毕业论文答辩演讲稿
2015/06/23 职场文书
详解Oracle块修改跟踪功能
2021/11/07 Oracle
vue 自定义的组件绑定点击事件
2022/04/21 Vue.js