Take correct shape of transformed parameter.
@@ -123,6 +123,8 @@ class Parameter(tf.Module):
@property
def shape(self):
+ if self.transform is not None:
+ return self.transform.forward_event_shape(self._unconstrained)
return self._unconstrained.shape
def get_shape(self):
