use x0.size instead of x0.shape[0]

tttamaki · tttamaki · commit 4fae288779bb · 2026-02-27T16:29:50.000+09:00
diff --git a/pyproximal/optimization/primal.py b/pyproximal/optimization/primal.py
@@ -2063,7 +2063,7 @@ def ConsensusADMM(  # pylint: disable=invalid-name
     m = len(proxfs)
     x_bar = x0.copy()
     x_bar_old = x0.copy()
-    y = ncp.zeros((m, x0.shape[0]))
+    y = ncp.zeros((m, x0.size))
 
     # iterate
     for iiter in range(niter):
