code: def to_numpy_squeeze(
      nest: types.NestedTensor, axis=0
  ) -> types.NestedArray:
      """Converts a nest of Tensors to a nest of numpy arrays and squeeze axis."""
      return tree.map_structure(
          lambda x: tf.squeeze(x, axis=axis).numpy(), nest
      )
  
imports: import functools
  from typing import List, Optional, Union
  from acme import types
  from acme.utils import tree_utils
  import sonnet
  import tensorflow
  import tree
label_desc: Use image node_id 10 for calling a global function with example usage: to_numpy_squeeze(nest, axis) and returns: tree