code: def get_activation(name):
      if name == "silu":
          return nn.SiLU
      elif name == "relu":
          return nn.ReLU
      elif name == "leaky_relu":
          return nn.LeakyReLU
      else:
          raise NotImplementedError(f"Unknown activation: {name}")
  
imports: import functools
  from functools import partial
  import numpy
  import torch
  from torch import Tensor, nn
  from torch.nn.modules.batchnorm import _BatchNorm
label_desc: Use image node_id 2 for calling a global function with example usage: get_activation(name) and returns: nn, nn, nn