code: def apex():
      for i in range(1000):
          unflat = unflatten_apex(flat_t, unflat_t)
  
file_path: DeepSpeed/tests/benchmarks/unflatten_bench.py
imports: import argparse
  import gc
  import torch
  from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
  from deepspeed.accelerator import get_accelerator
  from deepspeed.ops.op_builder import UtilsBuilder
  from apex_C import flatten
  from apex_C import unflatten
  import cProfile
  import timeit
label_desc: Use image node_id 3 for calling a global function with example usage: apex() without return types