code: def batch_remap_label_ids(labels: np.ndarray) -> np.ndarray:
      """Remaps a batch of labels so that the first occurrence of each label ID is in ascending order.
  
      This is a batched version of `remap_label_ids`. The mappings are independent
      between examples in the batch.
  
      Arguments:
        labels: A `np.ndarray` of shape `[batch_size, num_points]`.
  
      Returns:
        batch_remapped_labels: A `np.ndarray` of shape `[batch_size, num_points]`.
        batch_old_id_to_new_id: A list of dictionaries that map old cluster IDs to
          new cluster IDs.
      """
      batch_remap_output = [remap_label_ids(label) for label in labels]
      batch_remapped_labels = np.stack(
          [x[0] for x in batch_remap_output]
      )
      batch_old_id_to_new_id = [x[1] for x in batch_remap_output]
      return batch_remapped_labels, batch_old_id_to_new_id
  
label_desc: Use image node_id 2 for calling a global function with example usage: batch_remap_label_ids(labels) and returns: batch_remapped_labels,  batch_old_id_to_new_id