tensorplay.nn.functional.triplet_margin_with_distance_loss

tensorplay.nn.functional.triplet_margin_with_distance_loss(anchor: TensorBase, positive: TensorBase, negative: TensorBase, *, distance_function=None, margin: float = 1.0, swap: bool = False, reduction: str = 'mean') → TensorBase[source]

Compute the triplet margin loss using a custom distance function.

See TripletMarginWithDistanceLoss for details.