code: def __init__(self, size, dtype=chx.float32):
      shape = (size,)
      self.avg_mean = chx.zeros(shape, dtype)
      self.avg_var = chx.zeros(shape, dtype)
      self.gamma = chx.ones(shape, dtype)
      self.beta = chx.zeros(shape, dtype)
  
imports: import numpy
  import chainerx
label_desc: Use image node_id 1 to create a new BatchNormalization object from inherited base classes: object with example: obj = BatchNormalization(size, dtype)