match_flag_batch = [
          tf.constant([1.0, 1.0]),
          tf.constant([1.0]),
          tf.constant([1.0, 1.0]),