input_mask_repeat = Lambda(lambda x: K.repeat(x, config.EMB_SIZE), lambda x: x)(input_mask_float)