abstract.py 文件源码

def get_init_state(self, batch_size):
        '''
        Construct the initial state of the grammar state machine.

        Returns:
            A tensor of dtype tf.int32 with shape (batch_size,)
        '''
        return tf.zeros((batch_size,), dtype=tf.int32)