使用pytorch将标签转为one-hot向量

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

F_aF_a

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

阅读量2.8k

点赞数 2

分类专栏：机器学习 # pytorch Python 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_37252519/article/details/120234389

版权

Python 同时被 3 个专栏收录

42 篇文章 0 订阅

订阅专栏

pytorch

12 篇文章 0 订阅

订阅专栏

机器学习

9 篇文章 1 订阅

订阅专栏

文章目录

使用`nn.functional.one_hot`
使用`scatter_`
- 小样本

使用`nn.functional.one_hot`

print(torch.nn.functional.one_hot(labels, num_classes=5))

使用`scatter_`

https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html?highlight=scatter_#torch.Tensor.scatter_

小样本

n_way = 5
n_shot = 1
print(labels.shape) # torch.Size([5])
one_hot_labels  = torch.zeros(n_way * n_shot, n_way).scatter_(1, labels.view(-1, 1), 1)
print(one_hot_labels)
"""
tensor([[1., 0., 0., 0., 0.],
        [0., 1., 0., 0., 0.],
        [0., 0., 1., 0., 0.],
        [0., 0., 0., 1., 0.],
        [0., 0., 0., 0., 1.]])
"""