def METHOD_NAME():
    # Insert a zero in the consecutive ids
    # where zero appears in unique_ids.
    # id_is_zero has length 1.
    zero_id_ind = tf.cast(id_is_zero[0], tf.int32)
    ids_before = nonzero_consecutive_ids[:zero_id_ind]
    ids_after = nonzero_consecutive_ids[zero_id_ind:]
    return tf.concat([ids_before, [0], ids_after], axis=0)