Testing:
There are likely not enough unit tests, however, I have done further testing against a naive implementation in another branch. If needed I can write more unit tests. I will PR the other branch depending on issue #299.
Algorithm:
Uses quickselect for expected O(n) complexity, however, the pivot is not yet randomized.
ghost
commented
3 years ago
Added TopK-1 for CPU.
Testing: There are likely not enough unit tests, however, I have done further testing against a naive implementation in another branch. If needed I can write more unit tests. I will PR the other branch depending on issue #299.
Algorithm: Uses quickselect for expected O(n) complexity, however, the pivot is not yet randomized.