[Discussion] Support sort in mshadow?
Closed this issue · 4 comments
antinucleon commented
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
tqchen commented
if there is need for this, we can try to add a support.
antinucleon commented
The use case is very rare for now. What I know is:
- nms
- strange gates layer with some "experts" (doesn't work for 3 years research).
tqchen commented
then i will bring them as mxnet operator level first, before bring them into mshadow
antinucleon commented
LGTM