diff options
Diffstat (limited to 'ot/gpu/__init__.py')
| -rw-r--r-- | ot/gpu/__init__.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py new file mode 100644 index 0000000..40b11c0 --- /dev/null +++ b/ot/gpu/__init__.py @@ -0,0 +1,7 @@ +# -*- coding: utf-8 -*- + +from . import bregman +from . import da +from .bregman import sinkhorn + +__all__ = ["bregman", "da", "sinkhorn"] |