From 0b806374d33ae83d39846096a1838b096c0c0b8e Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Tue, 7 Feb 2017 17:17:34 +0100
Subject: log return for sinkhorn

---
 ot/bregman.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'ot/bregman.py')

diff --git a/ot/bregman.py b/ot/bregman.py
index fd22259..27f8ff5 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -139,6 +139,10 @@ def sinkhorn(a,b, M, reg, numItermax = 1000, stopThr=1e-9, verbose=False, log=Fa
                     print('{:5s}|{:12s}'.format('It.','Err')+'\n'+'-'*19)
                 print('{:5d}|{:8e}|'.format(cpt,err))
         cpt = cpt +1
+    if log:
+        log['u']=u
+        log['v']=v
+          
     #print 'err=',err,' cpt=',cpt
     if log:
         return u.reshape((-1,1))*K*v.reshape((1,-1)),log
-- 
cgit v1.2.3