CrossEntropyLoss — PyTorch 2.8 documentation