Rate this Page

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