pytorch中的一个热编码

我真的是一个新的编码，现在我正试图把我的标签一个热编码。我已经将np.array转换成张量，如下所示

张量（[4,4,4,4,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., 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）您必须将其转换为

long

类型。不能用浮球<代码>F.one_hot仅使用长传感器

F.one_hot(t.long())