code: def get_trans_fun(name):
      """Retrieves the transformation function by name."""
      trans_funs = {
          "basic_transform": BasicTransform,
          "bottleneck_transform": BottleneckTransform,
      }
      assert (
          name in trans_funs.keys()
      ), "Transformation function '{}' not supported".format(name)
      return trans_funs[name]
  
imports: import numpy
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: get_trans_fun(name) and returns: trans_funs