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][source]¶
Compute the triplet margin loss for input tensors using a custom distance function.
See
TripletMarginWithDistanceLoss
for details.- Return type