code: def testPmfShapeCountsPStretchedN(self):
      with self.cached_session():
          p = [0.1, 0.9]
          counts = [3.0, 2]
          n = np.full([4, 3], 5.0, dtype=np.float32)
          pmf = multinomial.Multinomial(total_count=n, probs=p).prob(
              counts
          )
          self.evaluate(pmf)
          self.assertEqual((4, 3), pmf.get_shape())
  
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 17 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testPmfShapeCountsPStretchedN() without return types