pytorch标签转onehot形式实例
2020-02-13 20:20
5146 查看
代码:
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形式实例就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- PyTorch——Tensor_把索引标签转换成one-hot标签表示
- pytorch: 如何优雅的将 int list 转成 one-hot形式
- Tensorflow将标签变为one-hot形式
- Keras中制作one-hot标签
- 02_Numpy的One-hot形式的转换(eye,identity)
- 多分类标签转换为one-hot向量
- tf.one_hot 实例
- TensorFlow 多分类标签转换成One-hot
- Struts2标签详解及具体实例解析
- 处理离散型特征和连续型特征共存的情况 归一化 论述了对离散特征进行one-hot编码的意义
- 从零开始-Machine Learning学习笔记(19)-One-hot与哑变量的区别与联系
- Thymeleaf 3.0 自定义标签方言属性,简单实例。快速学习!
- js实例,获取选中select标签的value和option text值
- 【php基础班】第3天 超链接a、table标签、表单、透明的flash实例
- 详解Dedecms各种形式的栏目调用标签,包括SQL调用方法
- transform实现HTML5 video标签视频比例拉伸实例详解
- jquery/js 实现 标签 条码 一维码 Web页面 区域打印解决方案 实例
- channelartlist标签调用实例
- hibernate one-to-one实例
- 一个用基于智能标签的生成HTML实例