code: def testMultinomialMean(self):
      with self.cached_session():
          n = 5.0
          p = [0.1, 0.2, 0.7]
          dist = multinomial.Multinomial(total_count=n, probs=p)
          expected_means = 5 * np.array(p, dtype=np.float32)
          self.assertEqual((3,), dist.mean().get_shape())
          self.assertAllClose(expected_means, dist.mean())
  
imports: import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops.distributions import multinomial
  from tensorflow.python.platform import test
label_desc: Use image node_id 18 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testMultinomialMean() without return types