pytorch自带的one-hot编码方法
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,
·
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、指定标签类别数时
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
21
0- 0
已为社区贡献4条内容
所有评论(0)