torch.Tensor.argsort

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

请参阅 torch.argsort()