code: def gen_rand_args(mod):
      args = []
      for _ in range(len(inspect.signature(mod.forward).parameters)):
          args.append(torch.randn(2, 3))
      return args
  
imports: import unittest
  from torch._lazy.ts_backend import init
  from torch._lazy import config
  from torch._lazy.extract_compiled_graph import extract_compiled_graph
  import torch
  from torch import nn
  import dis
  import inspect
  from torch import fx
  import re
  from contextlib import contextmanager
  import copy
label_desc: Use image node_id 3 for calling a global function with example usage: gen_rand_args(mod) and returns: args