from tensorflow.python.ops import array_ops

  @property
  def beta(self):
    """Scale parameter."""
    return self._beta

  def _batch_shape(self):
    return array_ops.broadcast_dynamic_shape(
        array_ops.shape(self.alpha), array_ops.shape(self.beta))

  def _get_batch_shape(self):
    return array_ops.broadcast_static_shape(
        self.alpha.get_shape(), self.beta.get_shape())

  def _event_shape(self):
