code: def testPdfXStretchedInBroadcastWhenSameRank(self):
      a = [[1.0, 2], [2.0, 3]]
      b = [[1.0, 2], [2.0, 3]]
      x = [[0.5, 0.5]]
      pdf = beta.Beta(a, b, validate_args=True).prob(x)
      self.assertAllClose(
          [[1.0, 3.0 / 2], [3.0 / 2, 15.0 / 8]],
          self.evaluate(pdf),
          rtol=1e-5,
      )
      self.assertEqual((2, 2), pdf.shape)
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import special
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 12 for calling the BetaTest obj's underlying member method code with example usage: obj.testPdfXStretchedInBroadcastWhenSameRank() without return types