Solved by KFrank in post #2 
              
              
                Hi Vikas! 

Yes.  Your output, the first argument to CrossEntropyLoss, being 
the output of your linear layer, are triples of values that range from 
-infinity to infinity, and can be understood as logits (in contrast to 
probabilities, that would range from zero to one).  Your labels are 
single va…