深度学习---one-hot向量

浪尽此生 提交于 2019-12-26 00:46:49

one-hot向量
为了将词表示为向量输入到神经网络,一个简单的办法是使用one-hot向量。假设词典中不同字符的数量为N,每个字符已经同一个从0到N-1的连续整数值索引一一对应。如果一个字符的索引是整数i,那么就创建一个全0的长为N的向量,并将其位置为i的元素设成1.该向量就是对原字符的one-hot向量。

label = torch.tensor([2,3,4,1]).view(-1,1)
one_hot = torch.zeros(4, 10).scatter_(1, label, 1)
print(one_hot)

tensor([[0., 0., 1., 0., 0., 0., 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., 1., 0., 0., 0., 0., 0., 0., 0., 0.]])

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!