class_docstr: RealNVP BatchNorm layer
class_docstr_tok: ["RealNVP","BatchNorm","layer"]
code: def __init__(self, input_size, momentum=0.9, eps=1e-5):
      super().__init__()
      self.momentum = momentum
      self.eps = eps
  
      self.log_gamma = nn.Parameter(torch.zeros(input_size))
      self.beta = nn.Parameter(torch.zeros(input_size))
  
      self.register_buffer("running_mean", torch.zeros(input_size))
      self.register_buffer("running_var", torch.ones(input_size))
  
imports: import copy
  import math
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.distributions import Normal
label_desc: Use image node_id 1 to create a new BatchNorm object from inherited base classes: nn with example: obj = BatchNorm(input_size, momentum, eps)