快速可区分的排序和排名
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_
评论0
最新资源