diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-07-24 14:31:45 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-07-24 14:31:45 +0200 |
commit | bbe411775b3d5abb5d6fb525262cccce3f73d345 (patch) | |
tree | 87c627bc59431e3abac711a1fe44d4ba896d6fdb /ot | |
parent | a228d145d0f435b2f2527c7a13437cafa98a4d65 (diff) |
test eisum instead of dot
Diffstat (limited to 'ot')
-rw-r--r-- | ot/bregman.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index ab84bcf..d2ade46 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -358,9 +358,9 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, while (err > stopThr and cpt < numItermax): uprev = u vprev = v - KtransposeU = np.dot(K.T, u) + KtransposeU = np.einsum('ij,i->j',K,u)#np.dot(K.T, u) v = np.divide(b, KtransposeU) - u = 1. / np.dot(Kp, v) + u = 1. / np.einsum('ij,j->i',Kp,v)#np.dot(Kp, v) if (np.any(KtransposeU == 0) or np.any(np.isnan(u)) or np.any(np.isnan(v)) or |