def _batch_shape(self):
    return functools.reduce(tf.broadcast_static_shape, (self.loc.shape, self.scale.shape, self.low.shape, self.high.shape))
def _batch_shape(self):
    return functools.reduce(tf.broadcast_static_shape, (self.loc.shape, self.scale.shape, self.low.shape, self.high.shape))