pytorch深度学习原理实战-rightness函数

在pytorch深度学习实战（集智俱乐部）书中卷积神经网络一章没有定义rightness函数。下面是自己实现代码

def rightness(output, target):
    preds = output.data.max(dim=1, keepdim=True)[1]
    return preds.eq(target.data.view_as(preds)).cpu().sum(), len(target)

@parameter
    output：模型输出
    target：标签
@return
     （正确样例数，总样本数量）