code: def unflatten(vector, shapes):
      i = 0
      arrays = []
      for shape in shapes:
          size = np.prod(shape, dtype=np.int_)
          array = vector[i : (i + size)].reshape(shape)
          arrays.append(array)
          i += size
      assert (
          len(vector) == i
      ), "Passed weight does not have the correct shape."
      return arrays
  
file_path: ray/python/ray/experimental/tf_utils.py
imports: from collections import OrderedDict, deque
  import numpy
  from ray.rllib.utils import force_list
  from ray.rllib.utils.framework import try_import_tf
label_desc: Use image node_id 1 for calling a global function with example usage: unflatten(vector, shapes) and returns: arrays