def get_init_state(self, batch_size): return tf.ones((batch_size,), dtype=tf.int32) * self.grammar.bookeeping_state_id