Solved by ptrblck in post #5 
              
              
                Sure, the model in the tutorial outputs 10 logits in its last linear layer. As you can see, no non-linearity was used on this layer, so that the values represent the raw logits for all 10 classes. 
If you call softmax on them, you would get the probabilities for each class, but we don’t want to do t…