diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 09:45:21 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 09:45:21 +0200 |
| commit | fa7f3ddbed0267edf634e359ce5b3a335807af3c (patch) | |
| tree | baf2551b8a88055572273a3b23e8c79c2144f988 | |
| parent | ee8ed4fa101861eec9e578f09aee4367af593af1 (diff) | |
correction import in ot.gpu
| -rw-r--r-- | ot/gpu/__init__.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index 9de2c40..0187a4f 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -1,8 +1,7 @@ # -*- coding: utf-8 -*- """ - -This module implement GPU ilmplementation for several OT solvers and utility +This module provides GPU implementation for several OT solvers and utility functions. The GPU backend in handled by `cupy <https://cupy.chainer.org/>`_. @@ -22,7 +21,7 @@ result of the function with parameter ``to_numpy=False``. from . import bregman from . import da from .bregman import sinkhorn -from .da +from .da import sinkhorn_lpl1_mm from . import utils from .utils import dist, to_gpu, to_np @@ -33,4 +32,5 @@ from .utils import dist, to_gpu, to_np # # License: MIT License -__all__ = ["utils", "dist", "sinkhorn", 'bregman', 'da', 'to_gpu', 'to_np'] +__all__ = ["utils", "dist", "sinkhorn", + "sinkhorn_lpl1_mm", 'bregman', 'da', 'to_gpu', 'to_np'] |