summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xWrappers/Python/ccpi/optimisation/functions/MixedL21Norm.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/Wrappers/Python/ccpi/optimisation/functions/MixedL21Norm.py b/Wrappers/Python/ccpi/optimisation/functions/MixedL21Norm.py
index a655e03..c5be084 100755
--- a/Wrappers/Python/ccpi/optimisation/functions/MixedL21Norm.py
+++ b/Wrappers/Python/ccpi/optimisation/functions/MixedL21Norm.py
@@ -99,8 +99,9 @@ class MixedL21Norm(Function):
else:
res1 = functools.reduce(lambda a,b: a + b*b, x.containers, x.get_item(0) * 0 )
res = res1.sqrt().maximum(1.0)
- for i,el in enumerate(x.containers):
- el.divide(res, out=out.get_item(i))
+ x.divide(res, out=out)
+ #for i,el in enumerate(x.containers):
+ # el.divide(res, out=out.get_item(i))
def __rmul__(self, scalar):
return ScaledFunction(self, scalar)