diff options
| author | LeoGautheron <leo_g_autheron@hotmail.fr> | 2018-07-16 06:56:34 +0200 |
|---|---|---|
| committer | LeoGautheron <leo_g_autheron@hotmail.fr> | 2018-07-16 06:56:34 +0200 |
| commit | 0764e356325df7e18f72c0ff468bfa8f8ee35059 (patch) | |
| tree | 1880c3c7c8fc0f044c7c68ef08a3d5a5e4352493 /ot/bregman.py | |
| parent | 73e61546b5509648ba6b7924d6e5a3ebe7fade5b (diff) | |
Add comment & fix flake8 error
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) |