Add files via upload

timlautk · Feb 6, 2018 · aa75b98 · aa75b98
1 parent c444d26
commit aa75b98
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Tools/relu_prox.m b/Tools/relu_prox.m
@@ -0,0 +1,8 @@
+function val = relu_prox(a,b,gamma,d,N)
+    val = zeros(d,N);
+    x = (a+gamma*b)/(1+gamma);
+    y = min(b,0);
+	val((a+gamma*b >= 0 & b >=0) | (a*(gamma-sqrt(gamma*(gamma+1))) <= gamma*b & b < 0)) = x((a+gamma*b >= 0 & b >=0) | (a*(gamma-sqrt(gamma*(gamma+1))) <= gamma*b & b < 0));
+    val(-a <= gamma*b & gamma*b <= a*(gamma-sqrt(gamma*(gamma+1)))) = b(-a <= gamma*b & gamma*b <= a*(gamma-sqrt(gamma*(gamma+1))));
+	val(a+gamma*b < 0) = y(a+gamma*b < 0);
+end