Hi everyone,
In this example:
for images, labels in test_loader:
    images = Variable(images).cuda()
    outputs = cnn(images)
    _, predicted = torch.max(outputs.data, 1)

I would like to know why we define the predictions by finding the maximum between outputs.data and 1.