code: def create_pyramid_split_indices(
      num_features_level0: int,
      num_pyramid_levels: int,
      feature_pyramid_downsample_stride: int,
      num_extra_features_per_level: int = 0,
  ):
      """Generates split indices for each pyramid level."""
      num_features_per_level = [
          (
              num_features_level0
              // (feature_pyramid_downsample_stride**idx)
          )
          + num_extra_features_per_level
          for idx in range(num_pyramid_levels)
      ]
      return np.cumsum(num_features_per_level)[:-1]
  
imports: from typing import Any, Dict, List, Mapping, Optional, Sequence, Union
  from absl import logging
  import flax
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.projects.vivit import model_utils
  from scenic.train_lib import train_utils
label_desc: Use image node_id 2 for calling a global function with example usage: create_pyramid_split_indices(num_features_level0, num_pyramid_levels, feature_pyramid_downsample_stride, num_extra_features_per_level) and returns: np