Solved by KFrank in post #10 
              
              
                Hello Chunchun! 


In general, there is no particular need to use probabilities to feed 
into your loss function. 
If your use case requires probabilities for some other reason, 
perhaps you could explain why you need them and what you 
need to use them for. 
For training, you should use (based on w…