diff options
-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'] |