code: def experience_to_transitions(
      experience: Trajectory, squeeze_time_dim: bool
  ) -> Transition:
      """Break experience to transitions."""
      transitions = to_transition(experience)
  
      if squeeze_time_dim:
          transitions = tf.nest.map_structure(
              lambda x: composite.squeeze(x, 1), transitions
          )
  
      return transitions
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import functools
  import pprint
  from typing import NamedTuple, Optional
  import numpy
  import tensorflow
  from tf_agents.trajectories import policy_step
  from tf_agents.trajectories import time_step
  from tf_agents.typing import types
  from tf_agents.utils import composite
  from tf_agents.utils import nest_utils
  from tf_agents.utils import value_ops
label_desc: Use image node_id 14 for calling a global function with example usage: experience_to_transitions(experience, squeeze_time_dim) and returns: transitions