diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2020-04-22 17:04:43 +0200 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2020-04-22 17:04:43 +0200 |
commit | f4d655bebbc3f4ea2919019cd5bc71bd0e8d426d (patch) | |
tree | cc1ec6dca4a8004bddebc0a7a84802ee72fb7c34 /src/python/gudhi/point_cloud/knn.py | |
parent | ba17759cf922d246a0a74ac5cf99f67d48a7d8c3 (diff) | |
parent | f218c8a94d3a383bad862dfcc2b92196ef245628 (diff) |
Merge remote-tracking branch 'origin/master' into wass-autodiff
Diffstat (limited to 'src/python/gudhi/point_cloud/knn.py')
-rw-r--r-- | src/python/gudhi/point_cloud/knn.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/python/gudhi/point_cloud/knn.py b/src/python/gudhi/point_cloud/knn.py index 4017e498..07553d6d 100644 --- a/src/python/gudhi/point_cloud/knn.py +++ b/src/python/gudhi/point_cloud/knn.py @@ -44,7 +44,8 @@ class KNearestNeighbors: Otherwise, neighbors are returned in an arbitrary order. Defaults to True. enable_autodiff (bool): if the input is a torch.tensor, jax.numpy.ndarray or tensorflow.Tensor, this instructs the function to compute distances in a way that works with automatic differentiation. - This is experimental and not supported for all metrics. Defaults to False. + This is experimental, not supported for all metrics, and requires the package EagerPy. + Defaults to False. kwargs: additional parameters are forwarded to the backends. """ self.k = k |