torch.Tensor.argsort

Tensor.argsort(dim=-1, descending=False) → LongTensor

See torch.argsort()