CrossEntropyLoss — PyTorch 2.4 documentation