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。

返回类型

Tensor