pytorch - pytorch 中的一种热门编码

Question

我对编码真的很陌生，现在我正在尝试将我的标签变成一种热门编码。我已经完成将 np.array 传输到张量，如下所示

tensor([4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 1., 1., 1.,
        1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 3., 3., 3., 3., 3., 3.,
        3., 3., 3., 3., 3., 3., 3., 3., 3., 0., 0., 0., 0., 0., 0., 0., 0., 0.,
        0., 0., 0., 0., 0., 0., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.,
        2., 2., 2.], dtype=torch.float64)

and I am using code to do one hot encoding 

aaa = F.one_hot(torch_qyh, num_classes=5)

但是，出现错误，显示“RuntimeError:one_hot 仅适用于索引张量”。任何帮助将不胜感激。

Answer 1

您必须将其转换为long 类型。不能用 float 来做到这一点。 F.one_hot 仅需要 LongTensor。

F.one_hot(t.long())