code: def unflatten_vector(vec, shapes):
      """
      reshape a flattened array
  
      :param vec: (np.ndarray) 1D arrays
      :param shapes: (tuple)
      :return: ([np.ndarray]) reshaped array
      """
      i = 0
      arrs = []
      for shape in shapes:
          size = np.prod(shape)
          arr = vec[i : i + size].reshape(shape)
          arrs.append(arr)
          i += size
      return arrs
  
file_path: stable-baselines/stable_baselines/common/math_util.py
imports: import numpy
  import scipy.signal
label_desc: Use image node_id 6 for calling a global function with example usage: unflatten_vector(vec, shapes) and returns: arrs