code: def get_activation_fn(activation: str) -> Callable:
      """Returns the activation function corresponding to `activation`"""
      if activation == "relu":
          return F.relu
      elif activation == "gelu":
          return gelu
      elif activation == "tanh":
          return torch.tanh
      elif activation == "linear":
          return lambda x: x
      else:
          raise RuntimeError(
              "--activation-fn {} not supported".format(activation)
          )
  
file_path: lightseq/lightseq/csrc/pytorch/util.py
imports: import math
  from typing import Callable
  import numpy
  import torch
  import torch.nn.functional
label_desc: Use image node_id 3 for calling a global function with example usage: get_activation_fn(activation) and returns: F, gelu, torch, function