CrossEntropyLoss — PyTorch 2.5 documentation