diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-07-24 14:34:17 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-07-24 14:34:17 +0200 |
| commit | 603c0eee29db890b0092ea8c848473bf413e186f (patch) | |
| tree | f32149937b9865e9e252a8e04b3a0721b6bcb42a /ot | |
| parent | a04112c69a62182c061d4b65e71ebb43c866d3e1 (diff) | |
pb index
Diffstat (limited to 'ot')
| -rw-r--r-- | ot/bregman.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 29ca9fd..57cedb2 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -359,7 +359,7 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, uprev = u vprev = v if nbb: - KtransposeU = np.einsum('ij,i,k->jk',K,u)#np.dot(K.T, u) + KtransposeU = np.einsum('ij,ik->jk',K,u)#np.dot(K.T, u) v = np.divide(b, KtransposeU) u = 1. / np.einsum('ij,jk->ik',Kp,v)#np.dot(Kp, v) else: |