torch.nn.functional.triplet_margin_loss
-
torch.nn.functional.triplet_margin_loss(anchor, positive, negative, margin=1.0, p=2, eps=1e-06, swap=False, size_average=None, reduce=None, reduction='mean')[source][source]
计算给定输入张量之间的三元组损失,并设置一个大于 0 的边界。
详细内容请见 TripletMarginLoss
。
- 返回类型:
张量