diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2019-06-25 14:36:23 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2019-06-25 14:36:23 +0200 |
| commit | c9775bde23527f8b55db53020352969405f8b6b5 (patch) | |
| tree | 4195d28c2c144f66d758b20baf862e9f32111cd9 /ot/bregman.py | |
| parent | 830d4ebd2e2c85b4f3503f358bb31a07918a27c5 (diff) | |
| parent | 2364d56aad650d501753cc93a69ea1b8ddf28b0a (diff) | |
Merge branch 'master' into doc_modules
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() |