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的Django框架中的项目进行单元测试的方法
Apr 11 Python
Python中shutil模块的常用文件操作函数用法示例
Jul 05 Python
通过Py2exe将自己的python程序打包成.exe/.app的方法
May 26 Python
使用Python写一个量化股票提醒系统
Aug 22 Python
利用python脚本如何简化jar操作命令
Feb 24 Python
Django如何简单快速实现PUT、DELETE方法
Jul 24 Python
python 串口读取+存储+输出处理实例
Dec 26 Python
python3实现在二叉树中找出和为某一值的所有路径(推荐)
Dec 26 Python
python实现飞船游戏的纵向移动
Apr 24 Python
如何验证python安装成功
Jul 06 Python
python 简单的调用有道翻译
Nov 25 Python
给numpy.array增加维度的超简单方法
Jun 02 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+MSSQL分页的例子
2006/10/09 PHP
比较简单的百度网盘文件直链PHP代码
2013/03/24 PHP
PHP设计模式之观察者模式实例
2016/02/22 PHP
PHP分页初探 一个最简单的PHP分页代码的简单实现
2016/06/21 PHP
PHP读取zip文件的方法示例
2016/11/17 PHP
超级有用的13个基于jQuery的内容滚动插件和教程
2011/07/31 Javascript
优化innerHTML操作(提高代码执行效率)
2011/08/20 Javascript
页面右下角弹出提示框示例代码js版
2013/08/02 Javascript
使用POST方式弹出窗口的两种方法示例介绍
2014/01/29 Javascript
jquery实现不包含当前项的选择器实例
2015/06/25 Javascript
Javascript之Date对象详解
2016/06/07 Javascript
Bootstarp 基础教程之表单部分实例代码
2017/02/03 Javascript
原生JS实现九宫格抽奖效果
2017/04/01 Javascript
返回上一个url并刷新界面的js代码
2020/09/12 Javascript
原生JavaScript实现拖动校验功能
2020/09/29 Javascript
Vue时间轴 vue-light-timeline的用法说明
2020/10/29 Javascript
Python完全新手教程
2007/02/08 Python
Python学生信息管理系统修改版
2018/03/13 Python
对Python2与Python3中__bool__方法的差异详解
2018/11/01 Python
详解Django+uwsgi+Nginx上线最佳实战
2019/03/14 Python
python 将字符串完成特定的向右移动方法
2019/06/11 Python
django orm模块中的 is_delete用法
2020/05/20 Python
Python pandas对excel的操作实现示例
2020/07/21 Python
Python爬取豆瓣数据实现过程解析
2020/10/27 Python
HTML5的Geolocation地理位置定位API使用教程
2016/05/12 HTML / CSS
广州品高软件.net笔面试题目
2012/04/18 面试题
会计出纳岗位职责
2013/12/25 职场文书
物理力学求职信
2014/02/18 职场文书
医师定期考核实施方案
2014/05/07 职场文书
护理见习报告范文
2014/11/03 职场文书
2014年心理健康教育工作总结
2014/12/06 职场文书
公司老总年会致辞
2015/07/30 职场文书
解决numpy数组互换两行及赋值的问题
2021/04/17 Python
JS Canvas接口和动画效果大全
2021/04/29 Javascript
mysql5.7的安装及Navicate长久免费使用的实现过程
2021/11/17 MySQL
浅谈GO中的Channel以及死锁的造成
2022/03/18 Golang