pytorch标签转onehot形式实例

更新时间：2020年01月02日 09:12:06 作者：头发光了你就强了

今天小编就为大家分享一篇pytorch标签转onehot形式实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

代码：

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形式实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

浅谈pycharm导入pandas包遇到的问题及解决
这篇文章主要介绍了浅谈pycharm导入pandas包遇到的问题及解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-06-06
Python在OpenCV里实现极坐标变换功能
这篇文章主要介绍了在OpenCV里实现极坐标变换功能，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-09-09
Python 中如何使用 virtualenv 管理虚拟环境
这篇文章主要介绍了Python 中使用 virtualenv 管理虚拟环境的方法，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2021-01-01
解决python递归函数及递归次数受到限制的问题
这篇文章主要介绍了解决python递归函数及递归次数受到限制的问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2022-06-06
python对日志进行处理的实例代码
本篇文章给大家分享了关于python处理日志的方法以及相关实例代码，有兴趣的朋友们学习下。
2018-10-10
Django如何实现防止XSS攻击
这篇文章主要介绍了Django如何实现防止XSS攻击,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-10-10
python3解析库lxml的安装与基本使用
lxml是python的一个解析库,支持HTML和XML的解析,支持XPath解析方式，下面这篇文章主要给大家介绍了关于python3解析库lxml的安装与使用的相关资料，文中通过示例代码介绍的非常详细，需要的朋友可以参考下
2018-06-06
记一次python 内存泄漏问题及解决过程
最近工作中慢慢开始用python协程相关的东西，所以用到了一些相关模块，如aiohttp, aiomysql, aioredis等，用的过程中也碰到的很多问题，这里整理了一次内存泄漏的问题。分享到脚本之家平台，需要的朋友参考下
2018-11-11
使用PM2+nginx部署python项目的方法示例
这篇文章主要介绍了使用PM2+nginx部署python项目的方法示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-11-11
python3将视频流保存为本地视频文件
这篇文章主要为大家详细介绍了python3将视频流保存为本地视频文件，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-06-06