Fixes the "nonzero is not supported for tensors with more than INT_MAX elements" in get_all_triplets_indices#689

KevinMusgrave merged 2 commits intoKevinMusgrave:devKevinMusgrave/pytorch-metric-learning:devfrom

mkmenta:remove-big-tensor-nonzeromkmenta/pytorch-metric-learning:remove-big-tensor-nonzeroCopy head branch name to clipboard