Pytorch triplet margin loss .