PyTorch——Tensor_把索引标签转换成one-hot标签表示

最新推荐文章于 2024-01-11 01:55:37 发布

Vic时代

最新推荐文章于 2024-01-11 01:55:37 发布

阅读量6.1w

点赞数 17

分类专栏：深度学习框架Torch

本文链接：https://blog.csdn.net/VictoriaW/article/details/72874637

版权

深度学习框架Torch 专栏收录该内容

22 篇文章 28 订阅

订阅专栏

对于分类问题，标签可以是类别索引值也可以是one-hot表示。以10类别分类为例，lable=[3] 和label=[0, 0, 0, 1, 0, 0, 0, 0, 0, 0]是一致的.

现在给定索引标签，怎么将其转换为one-hot标签表示？

>>>class_num = 10
>>>batch_size = 4
>>>label = torch.LongTensor(batch_size, 1).random_() % class_num
 3
 0
 0
 8

>>>one_hot = torch.zeros(batch_size, class_num).scatter_(1, label, 1)
    0     0     0     1     0     0     0     0     0     0
    1     0     0     0     0     0     0     0     0     0
    1     0     0     0     0     0     0     0     0     0
    0     0     0     0     0     0     0     0     1     0