Hi
As far as I remember, we do not have cross entropy loss Function, whose inputs
can be interpreted as a probability distribution.
Although it has an overhead slightly, we can calculate independently the probability distribution
and the cross entropy from the output of the last layer with F.softmax() and F.softmax_cross_entropy(), respectively.
Best