code: def get_stem_func(name):
      """
      Retrieves the stem module by name.
      """
      trans_funcs = {"x3d_stem": X3DStem, "basic_stem": ResNetBasicStem}
      assert (
          name in trans_funcs.keys()
      ), "Transformation function '{}' not supported".format(name)
      return trans_funcs[name]
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 1 for calling a global function with example usage: get_stem_func(name) and returns: trans_funcs