diff options
| author | Mokhtar Z. Alaya <mzalaya@Mokhtars-iMac.local> | 2020-01-18 07:45:34 +0100 |
|---|---|---|
| committer | Mokhtar Z. Alaya <mzalaya@Mokhtars-iMac.local> | 2020-01-18 07:45:34 +0100 |
| commit | b3fb1ef40a482f0989686b79373060d764b62d38 (patch) | |
| tree | d93a3641d6ef2c21eb6a70c12770669274343263 /ot/bregman.py | |
| parent | 3be0c215143e16c59ddd3be902416e91c3292937 (diff) | |
clean
Diffstat (limited to 'ot/bregman.py')
| -rw-r--r-- | ot/bregman.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index aff9f8c..c304b5d 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -2117,10 +2117,11 @@ def screenkhorn(a, b, M, reg, ns_budget=None, nt_budget=None, uniform=False, res log['v'] = vsc_full log['Isel'] = Isel log['Jsel'] = Jsel + gamma = usc_full[:, None] * K * vsc_full[None, :] gamma = gamma / gamma.sum() if log: return gamma, log else: - return gamma
\ No newline at end of file + return gamma |