class_docstr: Numerically stable normalization layer.
class_docstr_tok: ["Numerically","stable","normalization","layer","."]
code: def init_weights_and_state(self, input_signature):
      self.state = running_mean_and_variance_init(
          input_signature.shape[2:]
      )
  
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 2 for calling the Normalize obj's underlying member method code with example usage: obj.init_weights_and_state(input_signature) without return types