@ptrblck
Thank you very much, this worked and makes perfect sense.
I’m just wondering, did I miss the documentation/example, is this somewhere described in detail? Also, I am guessing that nn.CrossEntropyLoss uses Softmax internally?
Again, many Thanks!