import tensorflow as tf

        answer_batch_unstack = tf.unstack(answer_batch, axis=1)

        # initialize all the variables
        state_t_1 = init_state
        context_t_1 = tf.zeros([batch_size, encoder_dim])
        coverage_t_1 = None

        # store variables from each time-step
