code: def build_norm(
      name: str = "bn2d",
      num_features: Optional[int] = None,
      **kwargs: Any,
  ) -> Optional[nn.Module]:
      if name in ["ln", "ln2d"]:
          kwargs["normalized_shape"] = num_features
      else:
          kwargs["num_features"] = num_features
      if name in REGISTERED_NORM_DICT:
          norm_cls = REGISTERED_NORM_DICT[name]
          args = build_kwargs_from_config(kwargs, norm_cls)
          return norm_cls(**args)
      else:
          return None
  
file_path: kornia/kornia/contrib/models/efficient_vit/nn/norm.py
imports: from __future__ import annotations
  from typing import Any, Optional
  import torch
  from torch import nn
  from kornia.contrib.models.efficient_vit.utils import build_kwargs_from_config
label_desc: Use image node_id 1 for calling a global function with example usage: build_norm(name, num_features) and returns: norm_cls, None