diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 16:40:06 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 16:40:06 +0200 |
commit | 897982718a5fd81a9a591d80a7d50839399fc088 (patch) | |
tree | 48189493c09cda25ee19dfd0b7ef59c2f6819ba7 /ot/bregman.py | |
parent | 50bc90058940645a13e2f3e41129bdc97161dc63 (diff) |
fix func names + add more tests
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() |