broadcast fix

PythonOT · Oct 18, 2023 · b3578f0 · b3578f0
1 parent 2236ce4
commit b3578f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ot/mapping.py b/ot/mapping.py
@@ -124,7 +124,7 @@ def nearest_brenier_potential_fit(X, V, X_classes=None, a=None, b=None, strongly
     if init_method == 'target':
         G_val = V
     else:  # Init G_val with barycentric projection
-        G_val = emd(a, b, dist(X, V)) @ V / a
+        G_val = emd(a, b, dist(X, V)) @ V / a.reshape(n, 1)
     phi_val = None
     log_dict = {
         'G_list': [],