CrossEntropyLoss — PyTorch 1.10.1 documentation
pytorch.org › torchCrossEntropyLoss¶ class torch.nn. CrossEntropyLoss (weight = None, size_average = None, ignore_index =-100, reduce = None, reduction = 'mean', label_smoothing = 0.0) [source] ¶ This criterion computes the cross entropy loss between input and target. It is useful when training a classification problem with C classes.