def _prox(self, alpha, input): return thresh.l1_proj(self.epsilon, input)
def _prox(self, alpha, input): with backend.get_device(input): return thresh.l1_proj(self.epsilon, input)