diff --git a/pyro/distributions/transforms/basic.py b/pyro/distributions/transforms/basic.py index 585bbdb877..80ac844c53 100644 --- a/pyro/distributions/transforms/basic.py +++ b/pyro/distributions/transforms/basic.py @@ -53,7 +53,7 @@ class LeakyReLUTransform(Transform): Bijective transform via the mapping :math:`y = \text{LeakyReLU}(x)`. """ domain = constraints.real - codomain = constraints.positive + codomain = constraints.real bijective = True sign = +1