快速软排序:快速可区分的排序和排名 源码

快速可区分的排序和排名 O(n log n)中的可微分排序和排序操作。 依存关系 NumPy 科学 Numba Tensorflow(可选) PyTorch(可选) TensorFlow示例 >> > import tensorflow as tf >> > from fast_soft_sort . tf_ops import soft_rank , soft_sort >> > values = tf . convert_to_tensor ([[ 5. , 1. , 2. ], [ 2. , 1. , 5. ]], dtype = tf . float64 ) >> > soft_