diff --git a/proxtorch/operators/tvl1_3d.py b/proxtorch/operators/tvl1_3d.py index f80d46a..d26e0e5 100644 --- a/proxtorch/operators/tvl1_3d.py +++ b/proxtorch/operators/tvl1_3d.py @@ -76,7 +76,7 @@ def gradient(self, x): gradients[d, ...] = F.pad( torch.diff(x, dim=d, n=1), pad=get_padding_tuple(d, x.dim()) ) - gradients[:-1] *= (1.0 - self.l1_ratio) + gradients[:-1] *= 1.0 - self.l1_ratio gradients[-1] = self.l1_ratio * x return gradients