code: def config_link(r):
      """This is a helper function decorator to link the config to the head."""
  
      def wrapper(f):
          f.config_template = r
          return f
  
      return wrapper
  
imports: import math
  import torch
  from omegaconf import DictConfig
  from torch import nn
  from torch.autograd import Variable
  from torch.distributions import Categorical
  from pytorch_tabular.models.common.heads import config
  from pytorch_tabular.utils import _initialize_layers, _linear_dropout_bn
label_desc: Use image node_id 1 for calling a global function with example usage: config_link(r) and returns: wrapper, f