Example #1
0
 def _prox(self, alpha, input):
     return thresh.l1_proj(self.epsilon, input)
Example #2
0
 def _prox(self, alpha, input):
     with backend.get_device(input):
         return thresh.l1_proj(self.epsilon, input)