pytorch中的value, top = prob.topk(1, dim=1, largest=True, sorted=True)

函数：

prob = F.softmax(logit, 1)#按行softmax,行和为1
value, top = prob.topk(1, dim=1, largest=True, sorted=True)

topk函数参考网址：https://www.jb51.net/article/177713.htm

具体的，第一个参数为1，表示最大的，2表示前两大，…，依次类推
第二个参数为1，表示选每行最大的
返回参数分别为最大值与对应的索引