code: def batchnorm(input_dim, **kwargs):
      """
      A helper function to create a :class:`~pyro.distributions.transforms.BatchNorm`
      object for consistency with other helpers.
  
      :param input_dim: Dimension of input variable
      :type input_dim: int
      :param momentum: momentum parameter for updating moving averages
      :type momentum: float
      :param epsilon: small number to add to variances to ensure numerical stability
      :type epsilon: float
  
      """
      bn = BatchNorm(input_dim, **kwargs)
      return bn
  
file_path: pyro/pyro/distributions/transforms/batchnorm.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from ..None import constraints
  from ..torch_transform import TransformModule
  from ..util import copy_docs_from
label_desc: Use image node_id 1 for calling a global function with example usage: batchnorm(input_dim) and returns: bn