diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2019-06-25 14:33:46 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-06-25 14:33:46 +0200 |
commit | 2364d56aad650d501753cc93a69ea1b8ddf28b0a (patch) | |
tree | a92d17f75aa2e595656ab5bb58aeb3c9af16df3e /ot/bregman.py | |
parent | 5a6b226de20624b51c2ff98bc30e5611a7a788c7 (diff) | |
parent | c9df24649d359b21280328f6bd580eb049cae3d3 (diff) |
Merge pull request #87 from hichamjanati/unbalanced-ot
[MRG] Add Unbalanced KL Wasserstein distance + barycenter
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 321712b..09716e6 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -241,7 +241,7 @@ def sinkhorn2(a, b, M, reg, method='sinkhorn', numItermax=1000, b = np.asarray(b, dtype=np.float64) if len(b.shape) < 2: - b = b.reshape((-1, 1)) + b = b[:, None] return sink() |