code: def _get_activation_module(activation: str) -> torch.nn.Module:
      if activation == "relu":
          return torch.nn.ReLU()
      elif activation == "gelu":
          return torch.nn.GELU()
      elif activation == "silu":
          return torch.nn.SiLU()
      else:
          raise ValueError(f"Unsupported activation {activation}")
  
imports: import math
  from typing import List, Optional, Tuple
  import torch
  from torchaudio.models.emformer import _EmformerAttention, _EmformerImpl, _get_weight_init_gains
label_desc: Use image node_id 1 for calling a global function with example usage: _get_activation_module(activation) and returns: torch, torch, torch