code: def flat_to_shape(flat_tensor, shapes):
      """Convert a flat tensor to a list of tensors with specified shapes.
  
      `flat_tensor` must have exactly the number of elements as stated in `shapes`.
      """
      numels = [shape.numel() for shape in shapes]
      return [
          flat.reshape(shape)
          for flat, shape in zip(
              flat_tensor.split(split_size=numels), shapes
          )
      ]
  
file_path: torchsde/torchsde/_core/misc.py
imports: import warnings
  import torch
label_desc: Use image node_id 14 for calling a global function with example usage: flat_to_shape(flat_tensor, shapes) and returns: unknown