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编码方法
猜你喜欢
转载自blog.csdn.net/stay_zezo/article/details/121931429
今日推荐
周排行