code: def format_tensor(tensor: jax.Array) -> str:
      shape = list(tensor.shape)
      dtype = utils.simple_dtype(tensor.dtype)
      return f"{dtype}{shape}"
  
file_path: dm-haiku/haiku/_src/integration/checkpoint_utils.py
imports: from collections.abc import Mapping
  from typing import Any
  import haiku
  from haiku._src import utils
  from haiku._src.integration import descriptors
  import jax
  import jax.numpy
label_desc: Use image node_id 1 for calling a global function with example usage: format_tensor(tensor) and returns: dtype, shape