Solved by KFrank in post #2 
              
              
                Hello Maria! 

Sure.  And if you use normal tensor operations, autograd 
will work for you. 
Let’s say you have a binary classification problem and 
logits are your predictions for a batch (running from 
-infinity to +infinity) and labels are your known class 
labels for that batch (equal to 0 or 1)…