code: def unflatten_tree(tree, xs):
      """Inverse operation of `flatten_tree`."""
      return tf.nest.pack_sequence_as(tree, xs)
  
imports: import functools
  import numpy
  import six
  import tensorflow.compat.v2
  from tensorflow.python.util import nest
label_desc: Use image node_id 3 for calling a global function with example usage: unflatten_tree(tree, xs) and returns: tf