Solved by ptrblck in post #2 
              
              
                Since you are using the logarithm on softmax, you will get numbers in [-inf, 0], since log(0)=-inf and log(1)=0. 
You could get the probabilities back by using torch.exp(output).