diff options
author | Leo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr> | 2017-04-20 15:05:53 +0200 |
---|---|---|
committer | Leo gautheron <gautheron@iv-cm-359.creatis.insa-lyon.fr> | 2017-04-20 15:05:53 +0200 |
commit | fb95157f867d749551fad9a4988eb3dd2813096e (patch) | |
tree | d2d6dacde5ecb66bbab5e4cc3f9e92dccea4852a /ot/gpu/__init__.py | |
parent | 9c5cc82b2e6f6dcd94a24a47f414c278c717ee7a (diff) |
more changes from feeback
in addition add the posibility to normalize the cost matrix through the function fit
Diffstat (limited to 'ot/gpu/__init__.py')
-rw-r--r-- | ot/gpu/__init__.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index 9319208..40b11c0 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -2,5 +2,6 @@ from . import bregman from . import da +from .bregman import sinkhorn -__all__ = ["bregman", "da"] +__all__ = ["bregman", "da", "sinkhorn"] |