`ConditionedPlanar` class has correct `event_dim`
@@ -14,6 +14,11 @@ from pyro.nn import DenseNN
@copy_docs_from(Transform)
class ConditionedPlanar(Transform):
+ domain = constraints.real
+ codomain = constraints.real
+ bijective = True
+ event_dim = 1
+
def __init__(self, bias=None, u=None, w=None):
super(ConditionedPlanar, self).__init__(cache_size=1)
self.bias = bias
