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获取外网ip地址的方法总结
Jul 02 Python
python更新列表的方法
Jul 28 Python
Python中getattr函数和hasattr函数作用详解
Jun 14 Python
请不要重复犯我在学习Python和Linux系统上的错误
Dec 12 Python
解决Pandas的DataFrame输出截断和省略的问题
Feb 08 Python
python开启debug模式的方法
Jun 27 Python
python 扩展print打印文件路径和当前时间信息的实例代码
Oct 11 Python
使用Python来做一个屏幕录制工具的操作代码
Jan 18 Python
Python实现点云投影到平面显示
Jan 18 Python
python3将变量写入SQL语句的实现方式
Mar 02 Python
pandas.DataFrame.drop_duplicates 用法介绍
Jul 06 Python
python sleep和wait对比总结
Feb 03 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实现文件下载(支持中文文名)
2013/12/04 PHP
ThinkPHP中使用ajax接收json数据的方法
2014/12/18 PHP
php约瑟夫问题解决关于处死犯人的算法
2015/03/23 PHP
js Html结构转字符串形式显示代码
2011/11/15 Javascript
原生js实现半透明遮罩层效果具体代码
2013/06/06 Javascript
js键盘上下左右键怎么触发function(实例讲解)
2013/12/14 Javascript
JS方法调用括号的问题探讨
2014/01/24 Javascript
jQuery判断元素是否存在的可靠方法
2014/05/06 Javascript
详解AngularJS中的依赖注入机制
2015/06/17 Javascript
js限制文本框只能输入中文的方法
2015/08/11 Javascript
js实现网页图片延时加载 提升网页打开速度
2016/01/26 Javascript
js实现图片切换(动画版)
2016/12/25 Javascript
基于Vue 服务端Cookies删除的问题
2018/09/21 Javascript
JavaScript"模拟事件"的注意要点详解
2019/02/13 Javascript
利用weixin-java-miniapp生成小程序码并直接返回图片文件流的方法
2019/03/29 Javascript
简单了解vue中的v-if和v-show的区别
2019/10/08 Javascript
JavaScript实现抖音罗盘时钟
2019/10/11 Javascript
Python strip lstrip rstrip使用方法
2008/09/06 Python
轻松掌握python设计模式之访问者模式
2016/11/18 Python
python实现堆排序的实例讲解
2020/02/21 Python
Numpy中ndim、shape、dtype、astype的用法详解
2020/06/14 Python
ABOUT YOU匈牙利:500个最受欢迎的时尚品牌
2019/07/19 全球购物
P/Invoke是什么
2015/07/31 面试题
高中军训感言400字
2014/02/24 职场文书
大学生两会学习心得体会
2014/03/10 职场文书
小学新学期寄语
2014/04/02 职场文书
旅游饭店管理专业自荐书
2014/06/28 职场文书
海洋科学专业求职信
2014/08/10 职场文书
2014城乡环境综合治理工作总结
2014/12/19 职场文书
档案工作个人总结
2015/03/03 职场文书
活动经费申请报告
2015/05/15 职场文书
六五普法心得体会2016
2016/01/21 职场文书
话题作文之财富(600字)
2019/12/03 职场文书
常用的Python代码调试工具总结
2021/06/23 Python
MySQL多表查询机制
2022/03/17 MySQL
拙作再改《我的收音机情缘》
2022/04/05 无线电