diff options
| -rw-r--r-- | ot/bregman.py | 1 | ||||
| -rw-r--r-- | ot/utils.py | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 55c44f6..c8e69ce 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -344,6 +344,7 @@ def sinkhorn_knopp(a, b, M, reg, numItermax=1000, # print(reg) + # Next 3 lines equivalent to K= np.exp(-M/reg), but faster to compute K = np.empty(M.shape, dtype=M.dtype) np.divide(M, -reg, out=K) np.exp(K, out=K) diff --git a/ot/utils.py b/ot/utils.py index 14cc805..bb21b38 100644 --- a/ot/utils.py +++ b/ot/utils.py @@ -76,6 +76,7 @@ def clean_zeros(a, b, M): b2 = b[b > 0] return a2, b2, M2 + def euclidean_distances(X, Y, squared=False): """ Considering the rows of X (and Y=X) as vectors, compute the |