Solved by KFrank in post #2 
              
              
                Hi Julio! 

sigmoid() should do what you want. 
A couple of points: 
I imagine that you have a Linear layer feeding your activation function. 
The Linear layer produces a set raw-score logits.  softmax() converts 
these to a set class probabilities that sum to one.  sigmoid() converts 
each logit in…