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 random模块常用方法
Nov 03 Python
python中黄金分割法实现方法
May 06 Python
Python smallseg分词用法实例分析
May 28 Python
Python中Django框架下的staticfiles使用简介
May 30 Python
200 行python 代码实现 2048 游戏
Jan 12 Python
python gensim使用word2vec词向量处理中文语料的方法
Jul 05 Python
pandas DataFrame行或列的删除方法的实现示例
Aug 02 Python
python实现登录密码重置简易操作代码
Aug 14 Python
TensorBoard 计算图的查看方式
Feb 15 Python
Python常用编译器原理及特点解析
Mar 23 Python
Jmeter调用Python脚本实现参数互相传递的实现
Jan 22 Python
python数字类型和占位符详情
Mar 13 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 命令行工具 shell_exec, exec, passthru, system详细使用介绍
2011/09/11 PHP
php数组函数array_walk用法示例
2016/05/26 PHP
JS关闭窗口或JS关闭页面的几种代码分享
2013/10/25 Javascript
使用jquery获取url以及jquery获取url参数的实现方法
2016/05/25 Javascript
返回函数的JavaScript函数
2016/06/14 Javascript
jQuery ajax MD5实现用户注册即时验证功能
2016/10/11 Javascript
详解vue.js的事件处理器v-on:click
2017/06/27 Javascript
angular学习之从零搭建一个angular4.0项目
2017/07/10 Javascript
移动端Ionic App 资讯上下循环滚动的实现代码(跑马灯效果)
2017/08/29 Javascript
使用Vue.js开发微信小程序开源框架mpvue解析
2018/03/20 Javascript
Vue中 v-if/v-show/插值表达式导致闪现的原因及解决办法
2018/10/12 Javascript
d3.js实现图形拖拽
2019/12/19 Javascript
从零使用TypeScript开发项目打包发布到npm
2020/02/14 Javascript
vuex(vue状态管理)的特殊应用案例分享
2020/03/03 Javascript
python thread 并发且顺序运行示例
2009/04/09 Python
php使用递归与迭代实现快速排序示例
2014/01/23 Python
python开发之for循环操作实例详解
2015/11/12 Python
Python 2与Python 3版本和编码的对比
2017/02/14 Python
Python爬虫实例扒取2345天气预报
2018/03/04 Python
解决python3中cv2读取中文路径的问题
2018/12/05 Python
使用python进行波形及频谱绘制的方法
2019/06/17 Python
python实现超市商品销售管理系统
2019/11/22 Python
HTML5 visibilityState属性详细介绍和使用实例
2014/05/03 HTML / CSS
YSL圣罗兰美妆俄罗斯官网:Yves Saint Lauret RU
2020/09/23 全球购物
安全教育心得体会
2013/12/29 职场文书
四年级语文教学反思
2014/02/05 职场文书
奶茶店创业计划书
2014/08/14 职场文书
小学生2014国庆节演讲稿:祖国在我心中
2014/09/21 职场文书
2014年民警工作总结
2014/11/25 职场文书
2014年纳税评估工作总结
2014/12/23 职场文书
银行催款通知书
2015/04/17 职场文书
小学入学感言
2015/08/01 职场文书
车位出租协议书范本
2016/03/19 职场文书
Redis高级数据类型Hyperloglog、Bitmap的使用
2021/05/24 Redis
springboot 多数据源配置不生效遇到的坑及解决
2021/11/17 Java/Android
5人制售《绝地求生》游戏外挂获利500多万元 被判刑
2022/03/31 其他游戏