整理：TripletMarginLoss、nn.MarginRankingLoss

在pytorch中，提供了两个损失函数，都与triplet loss相关。但是使用的方式不一样。

 

一、TripletMarginLoss

这个就是最正宗的Triplet Loss的实现。它的输入是anchor, positive, negative三个B*N的张量（表示Batchsize个N为的特征向量），输出triplet loss的值。

定义为：

criterion = torch.nn.TripletMarginLoss(margin=1.0, p=2.0, eps=1e-06, swap=False, size_average=None, reduce=None, reduction='mean')

参数：

margin (float) – 默认为1
p (int) – norm degree，默认为2
reduction (string) – 指定返回各损失值(none)，批损失均值(mean)，批损失和(sum)，默认返回批损失均值(mean)
 

使用：

loss = criterion(anchor, positive, negative)

输入三个二维张量，每个向量的尺寸：(B,N