pytorch自带的one-hot编码方法

pytorch现在自带的将标签转成one-hot编码方法

torch.nn.functional.one_hot(tensornum_classes=- 1) → LongTensor

下图是pytorch官网的例子

pytorch自带的one-hot编码方法_第1张图片

1、不指定标签类别数时,pytorch默认将tensor中最大值作为标签类别最大数

下面的tensor([2,3,4]), 标签类别最大数会默认为4, 即默认是有标签(0, 1, 2, ,3, 4)五中标签

 pytorch自带的one-hot编码方法_第2张图片

 2、指定标签类别数时

pytorch自带的one-hot编码方法_第3张图片

你可能感兴趣的:(深度学习,pytorch,人工智能,深度学习)