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 logging
  import math
  from typing import List
  import torch
  import torch.nn
  import torch.nn.functional
  from reagent.core import parameters, types
  from reagent.models import convolutional_network, fully_connected_network
  from reagent.models.base import ModelBase
  from reagent.models.fully_connected_network import ACTIVATION_MAP
label_desc: Use image node_id 1 for calling a global function with example usage: _get_activation_fn(activation) and returns: F, F