diff --git a/pyproximal/optimization/primal.py b/pyproximal/optimization/primal.py index f7a568a..8f01767 100644 --- 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_like(x0) + y = ncp.zeros((m, x0.size), dtype=x0.dtype) # iterate for iiter in range(niter):