import tensorflow as tf

        src = batch_of_data["feature"]
        if self._trg_lang_tag_position in ["src", "source"]:
            src = tf.concat([tf.expand_dims(batch_of_data["trg_lang"], axis=1), src], axis=1)
        if self._with_src_lang_tag:
            src = tf.concat([tf.expand_dims(batch_of_data["src_lang"], axis=1), src], axis=1)

        input_dict = {"src": src,
                      "src_length": deduce_text_length(src, self._multilingual_dp.meta["pad_id"],
