diff options
Diffstat (limited to 'ot/sliced.py')
| -rw-r--r-- | ot/sliced.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/sliced.py b/ot/sliced.py index cf2d3be..20891a4 100644 --- a/ot/sliced.py +++ b/ot/sliced.py @@ -147,6 +147,8 @@ def sliced_wasserstein_distance(X_s, X_t, a=None, b=None, n_projections=50, p=2, if projections is None: projections = get_random_projections(d, n_projections, seed, backend=nx, type_as=X_s) + else: + n_projections = projections.shape[1] X_s_projections = nx.dot(X_s, projections) X_t_projections = nx.dot(X_t, projections) |