Tensor_把索引标签转换成one-hot标签表示

对于分类问题,标签可以是类别索引值也可以是one-hot表示。以10类别分类为例,lable=[3] 和label=[0, 0, 0, 1, 0, 0, 0, 0, 0, 0]是一致的.

现在给定索引标签,怎么将其转换为one-hot标签表示?

import torch
import torch.nn.functional as F
y = torch.tensor(3)
num_classes = torch.tensor(9)
y_oh = F.one_hot(y, num_classes)
print(y_oh)

# output = tensor([0, 0, 0, 1, 0, 0, 0, 0, 0])
上一篇:return,想要return多个数据怎么办?return单项,若要return多项,那么就return出去一个对象


下一篇:[elk]elasticsearch实现冷热数据分离