From 603c0eee29db890b0092ea8c848473bf413e186f Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Tue, 24 Jul 2018 14:34:17 +0200
Subject: pb index

---
 ot/bregman.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'ot')

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:
-- 
cgit v1.2.3