diff options
Diffstat (limited to 'ot/stochastic.py')
-rw-r--r-- | ot/stochastic.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/stochastic.py b/ot/stochastic.py index f34154b..9912223 100644 --- a/ot/stochastic.py +++ b/ot/stochastic.py @@ -407,6 +407,6 @@ def transportation_matrix_entropic(a, b, M, reg, method, numItermax=10000, return None opt_u = c_transform_entropic(b, M, reg, opt_v) - pi = (np.exp((opt_u[:, None] + opt_v[None, :] - M[:, :]) / reg) - * a[:, None] * b[None, :]) + pi = (np.exp((opt_u[:, None] + opt_v[None, :] - M[:, :]) / reg) * + a[:, None] * b[None, :]) return pi |