code: def _reflect_index_fixer(index: Array, size: int) -> Array:
      return jnp.floor_divide(
          _mirror_index_fixer(2 * index + 1, 2 * size + 1) - 1, 2
      )
  
imports: from collections.abc import Sequence
  import functools
  import itertools
  import operator
  import textwrap
  from typing import Callable
  import scipy.ndimage
  from jax._src import api
  from jax._src import util
  from jax import lax
  import jax.numpy
  from jax._src.numpy.util import _wraps
  from jax._src.typing import ArrayLike, Array
  from jax._src.util import safe_zip
label_desc: Use image node_id 4 for calling a global function with example usage: _reflect_index_fixer(index, size) and returns: jnp