函数:
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,表示选每行最大的
返回参数分别为最大值与对应的索引
函数:
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,表示选每行最大的
返回参数分别为最大值与对应的索引