diff options
Diffstat (limited to 'ot/gpu/bregman.py')
| -rw-r--r-- | ot/gpu/bregman.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py index 3fdf11b..4d4a8b7 100644 --- a/ot/gpu/bregman.py +++ b/ot/gpu/bregman.py @@ -6,7 +6,7 @@ Bregman projections for regularized OT with GPU import numpy as np -def sinkhornGPU(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False, +def sinkhorn(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False, log=False, cudamat=None): # init data Nini = len(a) |