Update ot/lp/dmmot.py

Store input variable instead of copying it
PythonOT · Aug 3, 2023 · 24a69c0 · 24a69c0
1 parent b370202
commit 24a69c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ot/lp/dmmot.py b/ot/lp/dmmot.py
@@ -137,7 +137,7 @@ def dmmot_monge_1dgrid_loss(A, verbose=False, log=False):
     """
 
     nx = get_backend(A)
-    A_copy = nx.copy(A)
+    A_copy = A
     A = nx.to_numpy(A)
 
     AA = [np.copy(A[:, j]) for j in range(A.shape[1])]