https://www.tensorflow.org/versions/r2.15/api_docs/python/tf/argsort?authuser=0
tf.argsort | TensorFlow v2.15.0.post1
Returns the indices of a tensor that give its sorted order along an axis.
tfargsorttensorflowv215
https://arxiv.org/abs/2006.16038
[2006.16038] SoftSort: A Continuous Relaxation for the argsort Operator
Abstract page for arXiv paper 2006.16038: SoftSort: A Continuous Relaxation for the argsort Operator
continuous relaxationfor the2006argsortoperator
https://www.tensorflow.org/probability/api_docs/python/tfp/experimental/distributions/marginal_fns/ps/argsort
tfp.experimental.distributions.marginal_fns.ps.argsort | TensorFlow Probability
Returns the indices of a tensor that give its sorted order along an axis.
tfpexperimentaldistributionsmarginalfns