code: def _get_norm(norm_type):
      if norm_type == "CbN":
          return ComplexBatchNorm
      elif norm_type == "bN":
          return partial(OnReIm, BatchNorm)
      else:
          raise NotImplementedError(f"Unknown norm type: {norm_type}")
  
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 7 for calling a global function with example usage: _get_norm(norm_type) and returns: ComplexBatchNorm, partial