original_docstring: Static batch shape of models represented by this component.
  
      Returns:
        batch_shape: A `tf.TensorShape` giving the broadcast batch shape of
          all model parameters. This should match the batch shape of
          derived state space models, i.e.,
          `self.make_state_space_model(...).batch_shape`. It may be partially
          defined or unknown.
      
docstring: Static batch shape of models represented by this component.
docstring_tokens: Static batch shape of models represented by this component .
code: def batch_shape(self):
      """Static batch shape of models represented by this component.
  
      Returns:
        batch_shape: A `tf.TensorShape` giving the broadcast batch shape of
          all model parameters. This should match the batch shape of
          derived state space models, i.e.,
          `self.make_state_space_model(...).batch_shape`. It may be partially
          defined or unknown.
      """
      batch_shape = tf.TensorShape([])
      for param in self.parameters:
        batch_shape = tf.broadcast_static_shape(
            batch_shape, param.prior.batch_shape)
      return batch_shape
code_tokens: def batch_shape ( self ) : batch_shape = tf . TensorShape ( [ ] ) for param in self . parameters : batch_shape = tf . broadcast_static_shape ( batch_shape , param . prior . batch_shape ) return batch_shape
short_docstring: Static batch shape of models represented by this component.
short_docstring_tokens: Static batch shape of models represented by this component .
comment: """Static batch shape of models represented by this component.
  
      Returns:
        batch_shape: A `tf.TensorShape` giving the broadcast batch shape of
          all model parameters. This should match the batch shape of
          derived state space models, i.e.,
          `self.make_state_space_model(...).batch_shape`. It may be partially
          defined or unknown.
      """