diff options
Diffstat (limited to 'ot/bregman')
| -rw-r--r-- | ot/bregman/sink.py (renamed from ot/bregman/sinkhorn.py) | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman/sinkhorn.py b/ot/bregman/sink.py index 798ac97..8b97e1e 100644 --- a/ot/bregman/sinkhorn.py +++ b/ot/bregman/sink.py @@ -59,7 +59,7 @@ def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9): #print reg - K = np.exp(-reg*M) + K = np.exp(-M/reg) #print np.min(K) Kp = np.dot(np.diag(1/a),K) |