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模块restful使用方法实例
Dec 10 Python
Python中多线程及程序锁浅析
Jan 21 Python
Python守护线程用法实例
Jun 23 Python
Python网络编程之TCP套接字简单用法示例
Apr 09 Python
python基于物品协同过滤算法实现代码
May 31 Python
python运行时强制刷新缓冲区的方法
Jan 14 Python
Django 多环境配置详解
May 14 Python
python求最大值最小值方法总结
Jun 25 Python
Python数据存储之 h5py详解
Dec 26 Python
Python random模块制作简易的四位数验证码
Feb 01 Python
浅析Python 序列化与反序列化
Aug 05 Python
python爬取新闻门户网站的示例
Apr 25 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正则过滤html标签、空格、换行符的代码(附说明)
2010/10/25 PHP
php学习笔记(三)操作符与控制结构
2011/08/06 PHP
php分割合并两个字符串的函数实例
2015/06/19 PHP
laravel model模型定义实现开启自动管理时间created_at,updated_at
2019/10/17 PHP
如何在Laravel之外使用illuminate组件详解
2020/09/20 PHP
JavaScript/Js脚本处理html元素的自定义属性解析(亲测兼容Firefox与IE)
2013/11/25 Javascript
简单的js图片轮换代码(js图片轮播)
2014/05/06 Javascript
js获取当前页面的url网址信息
2014/06/12 Javascript
兼容Firefox的Javascript XSLT 处理XML文件
2014/12/31 Javascript
浅析javascript中的事件代理
2015/11/06 Javascript
BootStrap智能表单实战系列(九)表单图片上传的支持
2016/06/13 Javascript
jQuery实现点击行选中或取消CheckBox的方法
2016/08/01 Javascript
移动端js图片查看器
2016/11/17 Javascript
Angular的模块化(代码分享)
2016/12/26 Javascript
canvas实现绘制吃豆鱼效果
2017/01/12 Javascript
jquery ajaxfileupload异步上传插件使用详解
2017/02/08 Javascript
jQuery使用jsonp实现百度搜索的示例代码
2020/07/08 jQuery
js canvas实现五子棋小游戏
2021/01/22 Javascript
[00:31]DOTA2荣耀之路7:Miracle-空血无敌斩
2018/05/31 DOTA
[47:22]Mineski vs Winstrike 2018国际邀请赛小组赛BO2 第二场 8.16
2018/08/17 DOTA
一个计算身份证号码校验位的Python小程序
2014/08/15 Python
python类继承用法实例分析
2014/10/10 Python
一个基于flask的web应用诞生 使用模板引擎和表单插件(2)
2017/04/11 Python
Python入门_学会创建并调用函数的方法
2017/05/16 Python
Python使用POP3和SMTP协议收发邮件的示例代码
2019/04/16 Python
学习Django知识点分享
2019/09/11 Python
一套Delphi的笔试题一
2016/02/14 面试题
北京捷通华声语音技术有限公司Java软件工程师笔试题
2012/04/10 面试题
商务专员岗位职责
2013/11/23 职场文书
校三好学生主要事迹
2014/01/11 职场文书
电钳工人个人求职信
2014/05/10 职场文书
毕业生代领毕业材料的授权委托书
2014/09/29 职场文书
工厂标语大全
2014/10/06 职场文书
python中super()函数的理解与基本使用
2021/08/30 Python
python模块与C和C++动态库相互调用实现过程示例
2021/11/02 Python
php实现自动生成验证码的实例讲解
2021/11/17 PHP