Solved by Nikronic in post #4 
              
              
                No, actually, softmax normalize logits between [0, 1] in the way that sum of all values adds up to 1, So, it can be interpreted as probability. And cross entropy does this internally. 
For instance, a output of nn.Linear can be like this for 2 classes: 
output = [[-2.3, 8.46],
          [-.56, 3.24]…