pytorch现在自带的将标签转成one-hot编码方法
torch.nn.functional.
one_hot
(tensor, num_classes=- 1) → LongTensor
下图是pytorch官网的例子
1、不指定标签类别数时,pytorch默认将tensor中最大值作为标签类别最大数
下面的tensor([2,3,4]), 标签类别最大数会默认为4, 即默认是有标签(0, 1, 2, ,3, 4)五中标签
2、指定标签类别数时
pytorch现在自带的将标签转成one-hot编码方法
torch.nn.functional.
one_hot
(tensor, num_classes=- 1) → LongTensor
下图是pytorch官网的例子
1、不指定标签类别数时,pytorch默认将tensor中最大值作为标签类别最大数
下面的tensor([2,3,4]), 标签类别最大数会默认为4, 即默认是有标签(0, 1, 2, ,3, 4)五中标签
2、指定标签类别数时