dmlc/mshadow

[Discussion] Support sort in mshadow?

Closed this issue · 4 comments

Do we need to do this? For CPU we can use std::sort, for GPU we can refer CUDA examples.

sampe_api

dst, idx = sort(src)

where idx is the argsort result

if there is need for this, we can try to add a support.

The use case is very rare for now. What I know is:

  1. nms
  2. strange gates layer with some "experts" (doesn't work for 3 years research).

then i will bring them as mxnet operator level first, before bring them into mshadow

LGTM