code: def inject_pytorch_nn():
      for name in dir(nn):
          if inspect.isclass(getattr(nn, name)) and issubclass(
              getattr(nn, name), nn.Module
          ):
              _original_classes[name] = getattr(nn, name)
              setattr(nn, name, getattr(nas_nn, name))
  
imports: import inspect
  import torch.nn
  import nni.nas.nn.pytorch.layers
label_desc: Use image node_id 2 for calling a global function with example usage: inject_pytorch_nn() without return types