• Adam Becker's avatar
    mixed changes · 95f6eda6
    Adam Becker 提交于
     - add multidim support for top_k
     - use unified TopKOp, can implement topk, argtopk, and both
    95f6eda6
sort.py 9.1 KB