pytorch scatter函数

最新推荐文章于 2023-11-04 23:48:08 发布

Egg_Hu

最新推荐文章于 2023-11-04 23:48:08 发布

阅读量206

点赞数

文章标签： one-hot编码 scatter()函数张量操作 PyTorch 标签转换

本文链接：https://blog.csdn.net/Egg_Hu/article/details/114534397

版权

scatter() 一般可以用来对标签进行 one-hot 编码，这就是一个典型的用标量来修改张量的一个例子

example：

class_num = 10
batch_size = 4
label = torch.LongTensor(batch_size, 1).random_() % class_num
#tensor([[6],
#        [0],
#        [3],
#        [2]])
torch.zeros(batch_size, class_num).scatter_(1, label, 1)
#tensor([[0., 0., 0., 0., 0., 0., 1., 0., 0., 0.],
#        [1., 0., 0., 0., 0., 0., 0., 0., 0., 0.],
#        [0., 0., 0., 1., 0., 0., 0., 0., 0., 0.],
#        [0., 0., 1., 0., 0., 0., 0., 0., 0., 0.]])