code: def _complex_elem_type(dtype: DTypeLike) -> DType:
      """Returns the float type of the real/imaginary parts of a complex dtype."""
      return np.abs(np.zeros((), dtype)).dtype
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  from functools import partial
  import re
  import textwrap
  from typing import Any, Callable, NamedTuple, TypeVar
  import warnings
  from jax._src import api
  from jax._src import config
  from jax._src import core
  from jax._src import dtypes
  from jax._src.lax import lax
  from jax._src.util import safe_zip, safe_map
  from jax._src.typing import Array, ArrayLike, DimSize, DType, DTypeLike, Shape
  import numpy
label_desc: Use image node_id 11 for calling a global function with example usage: _complex_elem_type(dtype) and returns: np