code: def running_mean_and_variance_init(shape):
      mean_state = running_mean_init(shape, fill_value=0.0)
      var_state = running_mean_init(shape, fill_value=1.0)
      return (mean_state, var_state)
  
file_path: trax/trax/rl/normalization.py
imports: import gin
  import numpy
  from trax import fastmath
  from trax import layers
label_desc: Use image node_id 5 for calling a global function with example usage: running_mean_and_variance_init(shape) and returns: mean_state,  var_state