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')[原始碼][原始碼]¶
使用自定義距離函數計算輸入張量的 triplet margin loss。
詳情請參閱
TripletMarginWithDistanceLoss
。- 回傳類型