At the start of the code, triplets are 2 dimensions. while executing the dimensions there are no items left in the tensor.
for example,
when I print triplet size I get following output:
torch.size([56,3])
torch.size([62,3])
torch.size([48,3])
torch.size([0])