code: def testPmfCountsStretchedInBroadcastWhenLowerRank(self):
      # The probabilities of one vote falling into class k is the mean for class
      # k.
      alpha = [[1.0, 2], [2.0, 3]]
      counts = [1.0, 0]
      pmf = dm.DirichletMultinomial(
          1.0, alpha, validate_args=True
      ).prob(counts)
      self.assertAllClose([1 / 3.0, 2 / 5.0], self.evaluate(pmf))
      self.assertAllEqual([2], pmf.shape)
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import dirichlet_multinomial
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 14 for calling the DirichletMultinomialTest obj's underlying member method code with example usage: obj.testPmfCountsStretchedInBroadcastWhenLowerRank() without return types