Cross Entropy Loss Pytorch Examples