code: def init_variables(self, din: int, rngs: nnx.Rngs):
      self.scale = nnx.Param(jax.numpy.ones((din,)))
      self.bias = nnx.Param(jax.numpy.zeros((din,)))
      self.mean = nnx.BatchStat(jax.numpy.zeros((din,)))
      self.var = nnx.BatchStat(jax.numpy.ones((din,)))
  
imports: import typing
  import jax
  import jax.numpy
  from jax import random
  from flax.experimental import nnx
label_desc: Use image node_id 5 for calling the BatchNorm obj's underlying member method code with example usage: obj.init_variables(din, rngs) without return types