code: def _get_activation_fn(activation):
      if activation == "relu":
          return F.relu
      elif activation == "gelu":
          return F.gelu
  
      raise RuntimeError(
          "activation should be relu/gelu, not {}".format(activation)
      )
  
imports: import math
  from typing import Optional, Tuple
  import torch
  import torch.nn.functional
  from torch.nn import Module, init
  from torch.nn.functional import dropout, linear, softmax
  from torch.nn.init import constant_, xavier_normal_, xavier_uniform_
  from torch.nn.modules.dropout import Dropout
  from torch.nn.modules.normalization import LayerNorm
  from torch.nn.parameter import Parameter
label_desc: Use image node_id 2 for calling a global function with example usage: _get_activation_fn(activation) and returns: F, F