torch.Tensor.topk

Tensor.topk(k, dim=None, largest=True, sorted=True)

See torch.topk()