Sparse Categorical Cross Entropy Pytorch