Pytorch 多分类结果测试
在模型训练过程中需要对当前的效果进行验证,或者训练结束后需要在测试集上对模型进行测试。比如多分类问题,网络的前向传播的结果是一个概率值Tensor,如果是一个10分类问题,并且batch=4,结果是一个4*10的Tensor,Tensor的每一行表示某张图片分别在10分类下的预测概率值。Pytorch中的argmax()函数可以返回Tensor中每一行最大值的索引,torch.eq()函数可以...