文件名称:快速软排序:快速可区分的排序和排名
文件大小:24KB
文件格式:ZIP
更新时间:2024-02-26 23:00:22
sorting tensorflow pytorch ranking differentiable
快速可区分的排序和排名 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_
【文件预览】:
fast-soft-sort-master
----README.md(3KB)
----tests()
--------pytorch_ops_test.py(2KB)
--------jax_ops_test.py(2KB)
--------tf_ops_test.py(3KB)
--------isotonic_test.py(1KB)
--------numpy_ops_test.py(7KB)
----fast_soft_sort()
--------pytorch_ops.py(4KB)
--------tf_ops.py(4KB)
--------third_party()
--------__init__.py(0B)
--------jax_ops.py(4KB)
--------numpy_ops.py(12KB)
----LICENSE(11KB)
----CONTRIBUTING.md(1KB)
----setup.py(1KB)