class_docstr: A BatchNorm layer used in several flows
class_docstr_tok: ["A","BatchNorm","layer","used","in","several","flows"]
code: def __init__(self, num_features, eps=1e-05, momentum=0.1):
      nn.Module.__init__(self)
  
      self.eps = eps
      self.momentum = momentum
  
      self.log_gamma = nn.Parameter(torch.zeros(num_features))
      self.beta = nn.Parameter(torch.zeros(num_features))
  
      self.register_buffer("running_mean", torch.zeros(num_features))
      self.register_buffer("running_var", torch.ones(num_features))
  
file_path: benchmark_VAE/src/pythae/models/normalizing_flows/layers.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from pythae.models.base.base_utils import ModelOutput
label_desc: Use image node_id 1 to create a new BatchNorm object from inherited base classes: nn with example: obj = BatchNorm(num_features, eps, momentum)