code: def get_states(self, n):
      
      return self.states[len(self.new_states):len(self.new_states) + n]
docs: Get the next n recurrent states.
  
      Called by layers in "incremental" mode.
  
      Args:
        n: an integer
      Returns:
        a list of n Tensors