CrossEntropyLoss — PyTorch 2.11 documentation