torch.nn.functional.binary_cross_entropy
function binary_cross_entropy(input: Tensor, target: Tensor, options?: BinaryCrossEntropyFunctionalOptions): TensorBinary cross entropy loss. Measures the binary cross entropy between input probabilities and target.
Parameters
inputTensor- Tensor of probabilities in [0, 1]
targetTensor- Tensor of binary targets (0 or 1)
optionsBinaryCrossEntropyFunctionalOptionsoptional