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形式实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天小编就为大家分享一篇Python实现RGB与HSI颜色空间的互换方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
这篇文章主要为大家详细介绍了MySQL适配器PyMySQL的相关资料，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-09-09
在安装Anaconda之前,有的已经安装过一个Python版本了,但是又不想删除这个Python版本,该怎么办呢??这篇文章主要给大家介绍了关于保留已有python安装Anaconda的几种法推荐,需要的朋友可以参考下2023-12-12
深度学习的MNIST手写数字数据集识别方式(准确率99%,附代码)
这篇文章主要介绍了深度学习的MNIST手写数字数据集识别方式(准确率99%,附代码),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-06-06
之前是用的是typora来写的文章，最近typora最近开始收费了，所以就不想用了，于是找到了一个替代品MarkText。本文将介绍如何通过Python实现截图自动生成符合markdown的链接，感兴趣的可以了解一下2022-01-01
这篇文章主要介绍了Python映射拆分操作符用法,实例分析了Python映射拆分操作符**的使用技巧,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了Python pywin32实现word与Excel的处理，pywin32处理Word大多数用于格式转换，因为一般读写操作都可以借助python-docx实现，除非真的有特殊要求，但大部分企业对Wrod操作不会有太多复杂需求2022-08-08
近年来，随着技术的发展，机器学习和深度学习在金融资产量化研究上的应用越来越广泛和深入。目前，大量数据科学家在Kaggle网站上发布了使用机器学习/深度学习模型对股票、期货、比特币等金融资产做预测和分析的文章。本文就来看看如何用python预测股票交易信号2021-05-05
这篇文章主要介绍了python解压缩文件或文件夹两种常见方式,Python的zipfile和shutil模块提供了强大的文件和文件夹压缩与解压缩功能,zipfile模块适合精细控制,而shutil模块则快速简单,文中提供了详细的代码示例,需要的朋友可以参考下2025-04-04
这不是一篇关于Python异常的全面介绍的文章，这只是在学习Python异常后的一篇笔记式的记录和小结性质的文章2013-10-10
最新评论