对于分类问题,标签可以是类别索引值也可以是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])