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]
使用自定义距离函数计算输入张量的三元组边缘损失。
详细内容请见 TripletMarginWithDistanceLoss
。
- 返回类型:
张量