1. output.max(1)按行来寻找最大值的索引位置,返回该最大值[0]和它的索引位[1]。等价torch.max(output, 1)。type_as(labels)表示按照将output的数据类型转换为labels的数据类型。
preds = output.max(1)[1].type_as(labels)
2. preds.eq(labels)返回bool类型,preds.eq(labels).double()返回01,便于统计总命中数量
correct = preds.eq(labels).double()
3.求出预测正确的标签数量
correct = correct.sum()
4.预测正确的数量除以总数
result = correct / len(labels)