code: def testNegativeBinomialFromMeanDispersion(self):
      total_count = 5.0
      probs = np.repeat(0.2, 3).astype(np.float32)
      x = np.array([4.0, 5.0, 6.0], dtype=np.float32)
      mu = stats.nbinom.mean(total_count, 1 - probs)
      negbinom_mean_disp = negative_binomial.NegativeBinomial.experimental_from_mean_dispersion(
          mu, 1 / total_count, validate_args=True
      )
      expected_log_pmf = stats.nbinom.logpmf(
          x, n=total_count, p=1 - probs
      )
      log_pmf = negbinom_mean_disp.log_prob(x)
      self.assertAllClose(expected_log_pmf, self.evaluate(log_pmf))
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import negative_binomial
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 11 for calling the NegativeBinomialTest obj's underlying member method code with example usage: obj.testNegativeBinomialFromMeanDispersion() without return types