Given a model with softmax output:
ps = model.forward(images[img_idx,:])

How do you print the probabilities?
a) print(ps)
b) print(torch.softmax(ps))
c) print(ps.softmax())
d) print(torch.softmax(ps, dim=1))