BCEWithLogitsLoss - PyTorch - W3cubDocs
docs.w3cub.com › torchBCEWithLogitsLoss class torch.nn.BCEWithLogitsLoss(weight: Optional[torch.Tensor] = None, size_average=None, reduce=None, reduction: str = 'mean', pos_weight: Optional[torch.Tensor] = None) [source] This loss combines a Sigmoid layer and the BCELoss in one single class.