torch.nn.functional.triplet_margin_with_distance_loss

torch.nn.functional.triplet_margin_with_distance_loss(anchor, positive, negative, *, distance_function=None, margin=1.0, swap=False, reduction='mean') [source]

Compute the triplet margin loss for input tensors using a custom distance function.

See TripletMarginWithDistanceLoss for details.

Return type

Tensor