code: def test_runningmeanstd():
      for x1, x2, x3 in [
          (np.random.randn(3), np.random.randn(4), np.random.randn(5)),
          (
              np.random.randn(3, 2),
              np.random.randn(4, 2),
              np.random.randn(5, 2),
          ),
      ]:
          rms = RunningMeanStd(epsilon=0.0, shape=x1.shape[1:])
  
          x = np.concatenate([x1, x2, x3], axis=0)
          ms1 = [x.mean(axis=0), x.var(axis=0)]
          rms.update(x1)
          rms.update(x2)
          rms.update(x3)
          ms2 = [rms.mean, rms.var]
  
          np.testing.assert_allclose(ms1, ms2)
  
file_path: baselines/baselines/common/running_mean_std.py
imports: import tensorflow
  import numpy
  from baselines.common.tf_util import get_session
label_desc: Use image node_id 2 for calling a global function with example usage: test_runningmeanstd() without return types