code: def test_higher_rank_shape(self):
      running_mean = sample_stats.RunningMean.from_shape(shape=(5, 3))
      for sample in range(6):
          running_mean = running_mean.update(tf.ones((5, 3)) * sample)
      mean = self.evaluate(running_mean.mean)
      self.assertAllEqual(np.ones((5, 3)) * 2.5, mean)
  
imports: from absl.testing import parameterized
  import numpy
  import scipy.stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.experimental.stats import sample_stats
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import diagnostic
label_desc: Use image node_id 1 for calling the RunningMeanTest obj's underlying member method code with example usage: obj.test_higher_rank_shape() without return types