The mentioned classes indices might not be the right ones (this is determined by the target), but the predicted ones (with the highest probability).
If you want to get probability values, you could use F.softmax to get values in the range [0, 1].
However do not pass these values to the criterion. Use them just for debugging/printing purposes.