code: def get_all_class_obj(module, key_word=""):
      members = inspect.getmembers(module)
      rs = []
      module_name = None
      for name, obj in members:
          if inspect.isclass(obj):
              if "modules." + key_word in obj.__module__:
                  rs.append(obj)
                  # print(obj)
                  module_name = obj.__module__.split(".")[-1]
  
      return rs, module_name
  
file_path: FATE/python/fate/components/components/nn/hook_code/extract_torch_modules.py
imports: import inspect
  from torch.nn.modules import linear, activation, rnn, dropout, sparse, pooling, conv, transformer, batchnorm
  from torch.nn.modules import padding, pixelshuffle
  from torch.nn.modules import loss
label_desc: Use image node_id 1 for calling a global function with example usage: get_all_class_obj(module, key_word) and returns: rs,  module_name