code: def unbatch_nested_array(nested_array):
      return tf.nest.map_structure(
          lambda x: np.squeeze(x, 0), nested_array
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import numbers
  from typing import Optional, Text
  from absl import logging
  import numpy
  from six.moves import zip
  import tensorflow
  from tf_agents.typing import types
  from tf_agents.utils import composite
  import wrapt
  from tensorflow.python.util import nest
label_desc: Use image node_id 25 for calling a global function with example usage: unbatch_nested_array(nested_array) and returns: tf