diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 09:41:22 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-28 09:41:22 +0200 |
commit | ee8ed4fa101861eec9e578f09aee4367af593af1 (patch) | |
tree | 1996b7edaafe5f671a5275a37d6fd5cf7bf2051a /ot/gpu/__init__.py | |
parent | 5e7bfbcbc99ce5915873147677b434c0b1d10fc8 (diff) |
update documentation
Diffstat (limited to 'ot/gpu/__init__.py')
-rw-r--r-- | ot/gpu/__init__.py | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index de4825d..9de2c40 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -1,8 +1,28 @@ # -*- coding: utf-8 -*- +""" + + +This module implement GPU ilmplementation for several OT solvers and utility +functions. The GPU backend in handled by `cupy +<https://cupy.chainer.org/>`_. + +By default, the functions in this module accept and return numpy arrays +in order to proide drop-in replacement for the other POT function but +the transfer between CPU en GPU comes with a significant overhead. + +In order to get the best erformances, we recommend to given only cupy +arrays to the functions and desactivate the conversion to numpy of the +result of the function with parameter ``to_numpy=False``. + + + + +""" from . import bregman from . import da from .bregman import sinkhorn +from .da from . import utils from .utils import dist, to_gpu, to_np @@ -13,4 +33,4 @@ from .utils import dist, to_gpu, to_np # # License: MIT License -__all__ = ["utils", "dist", "sinkhorn"] +__all__ = ["utils", "dist", "sinkhorn", 'bregman', 'da', 'to_gpu', 'to_np'] |