torch.Tensor.topk

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

参见 torch.topk()