Dev enhance sort ops (#3828)
* draft * test sort op * argsort * argmax * quick imple for top_k, to be tested * feat * test argmax * pass all tests * fix type * batch_axis_non_change true * final * format * refine by review * accurate axis check * Update oneflow/python/ops/transpose_util.py Co-authored-by: NYinggang Wang <wyg19970408@gmail.com> Co-authored-by: NMardino <42901638+MARD1NO@users.noreply.github.com> Co-authored-by: NYinggang Wang <wyg19970408@gmail.com> Co-authored-by: Noneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Former-commit-id: a29c228c
Showing
想要评论请 注册 或 登录