diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 12:09:15 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 12:09:15 +0200 |
commit | 75492827c89a47cbc6807d4859be178d255c49bc (patch) | |
tree | 197d5d420d71b265e6e0ce9134c572156db54ac0 /ot/gpu/bregman.py | |
parent | ff104a6dde2d652283f72d7901bbe79dfb8571ed (diff) |
add test sinkhorn
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 7881c65..2302f80 100644 --- a/ot/gpu/bregman.py +++ b/ot/gpu/bregman.py @@ -9,7 +9,7 @@ import cudamat def sinkhorn(a, b, M_GPU, reg, numItermax=1000, stopThr=1e-9, verbose=False, log=False, returnAsGPU=False): - """ + r""" Solve the entropic regularization optimal transport problem on GPU The function solves the following optimization problem: |