no_candy:CSDN认证博客专家
博客地址：https://blog.csdn.net/no_candy
torch.max()函数predic = torch.max(outputs.data, 1)[1].cpu().numpy()