A simple continuous relaxation for argsort improves performance and is easy to implement.
problem Discrete nature of argsort operator makes it unsuitable for gradient-based learning.
method Proposed a continuous relaxation for argsort operator that is simple, fast, and achieves state-of-the-art performance.
result The proposed relaxation achieves state-of-the-art performance and is faster than competing approaches.