[Numpy-discussion] EHN: Discusions about 'add numpy.topk'

David Menéndez Hurtado davidmenhur at gmail.com
Sat May 29 12:33:06 EDT 2021


On Sat, 29 May 2021, 4:29 pm Ralf Gommers, <ralf.gommers at gmail.com> wrote:

>
>
> On Fri, May 28, 2021 at 4:58 PM <kangkai at mail.ustc.edu.cn> wrote:
>
>> Hi all,
>>
>> Finding topk elements is widely used in several fields, but missed in
>> NumPy.
>> I implement this functionality named as  numpy.topk using core numpy
>> functions and open a PR:
>>
>> https://github.com/numpy/numpy/pull/19117
>>
>> Any discussion are welcome.
>>
>
> Thanks for the proposal Kang. I think this functionality is indeed a
> fairly obvious gap in what Numpy offers, and would make sense to add. A
> detailed comparison with other libraries would be very helpful here.
> TensorFlow and JAX call this function `top_k`, while PyTorch, Dask and
> MXNet call it `topk`.
>

When I saw `topk` I initially parsed it as "to pk", similar to the current
`tolist`. I think `top_k` is more explicit and clear.

/David
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mail.python.org/pipermail/numpy-discussion/attachments/20210529/179153a8/attachment.html>


More information about the NumPy-Discussion mailing list