summaryrefslogtreecommitdiff
path: root/ot/gpu/bregman.py
diff options
context:
space:
mode:
Diffstat (limited to 'ot/gpu/bregman.py')
-rw-r--r--ot/gpu/bregman.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py
index 2302f80..86bfec1 100644
--- a/ot/gpu/bregman.py
+++ b/ot/gpu/bregman.py
@@ -3,6 +3,11 @@
Bregman projections for regularized OT with GPU
"""
+# Author: Remi Flamary <remi.flamary@unice.fr>
+# Leo Gautheron <https://github.com/aje>
+#
+# License: MIT License
+
import numpy as np
import cudamat