diff options
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 55c44f6..c8e69ce 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -344,6 +344,7 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, # print(reg) + # Next 3 lines equivalent to K= np.exp(-M/reg), but faster to compute K = np.empty(M.shape, dtype=M.dtype) np.divide(M, -reg, out=K) np.exp(K, out=K) |