maxk = max((1,)) # 取top1准确率,若取top1和top5准确率改为max((1,5)) _, pred = output.topk(maxk, 1, True, True) topk参数中,maxk取得是top1准确率,dim=1是按行取值, largest=1是取最大值 结果如下, _ tensor([[ 0.2298], [-0.4794], [-0.1070], [-0.1785]]) pred tensor([[...