pytorch中的value, top = prob.topk(1, dim=1, largest=True, sorted=True)
函数:prob=F.softmax(logit,1)#按行softmax,行和为1value,top=prob.topk(1,dim=1,largest=True,sorted=True)topk函数参考网址:https://www.jb51.net/article/177713.htm具体的,第一个参数为1,表示最大的,2表示前两大,…,依次类推第二个参数为1,表示选每行最大的返回参数分别为最大