An initializer function for random normal coefficients.

def RandomNormalInitializer(stddev=1e-2):
  """An initializer function for random normal coefficients."""
  def init(shape, rng):
    return (stddev * backend.random.normal(rng, shape)).astype('float32')
  return init