Solved by KFrank in post #9 
              
              
                Hi Alice! 
Let me answer your question(s) two different ways. 

You could, but doing so would be overkill.  You can just call the 
function (or class) version of sigmoid() directly: 
my_logits = my_model (my_batch)
with torch.no_grad():
    my_probabilities = torch.nn.functional.sigmoid (my_logits)
…