torchsig.models.spectrogram_models.detr.criterion.batch_sigmoid_ce_loss

torchsig.models.spectrogram_models.detr.criterion.batch_sigmoid_ce_loss(inputs: Tensor, targets: Tensor)[source]
Parameters:
  • inputs – A float tensor of arbitrary shape. The predictions for each example.

  • targets

    A float tensor with the same shape as inputs. Stores the binary

    classification label for each element in inputs

    (0 for the negative class and 1 for the positive class).

Returns:

Loss tensor