評價此頁

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]#

使用自定義距離函式為輸入張量計算三元組邊距損失。

有關詳細資訊,請參閱 TripletMarginWithDistanceLoss

返回型別

張量