onehot

可以快速上手的开发文档

#onehot

pytorch实现onehot编码转为普通label标签

label转onehot的很多,但是onehot转label的有点难找,所以就只能自己实现以下,用的topk函数,..

pytorch标签转onehot形式实例

代码:importtorchclass_num=10batch_size=4label=torch.LongTensor(batch_size,1).random_()%class_numprint(label.size())o..