可以快速上手的开发文档
label转onehot的很多,但是onehot转label的有点难找,所以就只能自己实现以下,用的topk函数,..
代码:importtorchclass_num=10batch_size=4label=torch.LongTensor(batch_size,1).random_()%class_numprint(label.size())o..