def initial_state(self, first_layer_initial_state):
    initial_state = self.zero_state(BATCH_SIZE, tf.float32)
    initial_state = list(initial_state)
    initial_state[0] = first_layer_initial_state
    initial_state = tuple(initial_state)
    return initial_state
def initial_state(self, first_layer_initial_state):
    initial_state = self.zero_state(BATCH_SIZE, tf.float32)
    initial_state = list(initial_state)
    initial_state[0] = first_layer_initial_state
    initial_state = tuple(initial_state)
    return initial_state