Solved by KFrank in post #2 
              
              
                Hi Physics! 

Yes, the terminology in the documentation is somewhat unfortunate. 
The argument to the loss functions that pytorch refers to as input is 
indeed the “output” of the network (or derived from it).  I prefer to call 
this the “prediction.”  The loss function then compares the “prediction…