Imported from https://github.com/IcewineChen/mxnet-batch_hard_triplet_loss