code: def testPmfPStretchedInBroadcastWhenLowerRank(self):
      with self.cached_session():
          p = [0.1, 0.9]
          counts = [[1.0, 0], [0, 1]]
          pmf = multinomial.Multinomial(total_count=1.0, probs=p).prob(
              counts
          )
          self.assertAllClose([0.1, 0.9], self.evaluate(pmf))
          self.assertEqual((2), 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 13 for calling the MultinomialTest obj's underlying member method code with example usage: obj.testPmfPStretchedInBroadcastWhenLowerRank() without return types