torch.nn.functional.binary_cross_entropy_with_logits ...
pytorch.org › docs › stabletorch.nn.functional.binary_cross_entropy_with_logits. Function that measures Binary Cross Entropy between target and input logits. See BCEWithLogitsLoss for details. input – Tensor of arbitrary shape as unnormalized scores (often referred to as logits). weight ( Tensor, optional) – a manual rescaling weight if provided it’s repeated to ...
CrossEntropyLoss — PyTorch 1.10.1 documentation
pytorch.org › torchThe latter is useful for higher dimension inputs, such as computing cross entropy loss per-pixel for 2D images. The target that this criterion expects should contain either: Class indices in the range [ 0 , C − 1 ] [0, C-1] [ 0 , C − 1 ] where C C C is the number of classes; if ignore_index is specified, this loss also accepts this class ...