code: def _to_struct(x):
      if is_array(x):
          return jax.ShapeDtypeStruct(x.shape, x.dtype)
      else:
          return x
  
file_path: equinox/equinox/_eval_shape.py
imports: import functools
  from collections.abc import Callable
  from typing import Any, Union
  import jax
  import jax._src.traceback_util
  import jax.tree_util
  from jaxtyping import PyTree
  from ._caches import internal_lru_caches
  from ._filters import combine, is_array, partition
  from ._module import Static
label_desc: Use image node_id 3 for calling a global function with example usage: _to_struct(x) and returns: jax, x