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, 4)五中标签
2、指定标签类别数时
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
21
0- 0
已为社区贡献2条内容
所有评论(0)