torch.Tensor.topk

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

請參見 torch.topk()