HingeEmbeddingLossΒΆ
Measures the loss given an input tensor and a labels tensor (-1 or 1).
Abstract Signature:
HingeEmbeddingLoss(input: Tensor, target: Tensor, margin: float = 1.0, reduction: str = mean)
JAX (Core)
API:
optax.hinge_embedding_lossStrategy: Plugin (loss_reduction)