parameters: (self)
return_statement: return self._state_saver._received_batch_size
docstring: The batch_size of the given batch.
  
      Usually, this is the batch_size requested when initializing the SQSS, but
      if allow_small_batch=True this will become smaller when inputs are
      exhausted.
  
      Returns:
        A scalar integer tensor, the batch_size
docstring_summary: The batch_size of the given batch.
docstring_tokens: The batch_size of the given batch .
function: def batch_size(self):
      """The batch_size of the given batch.
  
      Usually, this is the batch_size requested when initializing the SQSS, but
      if allow_small_batch=True this will become smaller when inputs are
      exhausted.
  
      Returns:
        A scalar integer tensor, the batch_size
      """
      return self._state_saver._received_batch_size
function_tokens: def batch_size ( self ) : return self . _state_saver . _received_batch_size