code: def _full_flatten(xs):
      def flatten(x):
          return tf.reshape(x, shape=[-1])
  
      return tf.concat(
          tf.nest.flatten(tf.nest.map_structure(flatten, xs)), axis=-1
      )
  
file_path: probability/tensorflow_probability/python/internal/numerics_testing.py
imports: import functools
  from absl import logging
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 8 for calling a global function with example usage: _full_flatten(xs) and returns: tf, tf