code: def format_array(x: Any) -> str:
      """Formats the given array showing dtype and shape info."""
      return (
          simple_dtype(x.dtype)
          + "["
          + ",".join(map(str, x.shape))
          + "]"
      )
  
file_path: dm-haiku/haiku/_src/utils.py
imports: import collections.abc
  from collections.abc import Sequence
  import decimal
  import inspect
  import pprint
  import re
  from typing import Any, TypeVar, Union
  import jax
label_desc: Use image node_id 11 for calling a global function with example usage: format_array(x) and returns: unknown