code: def testPdfXStretchedInBroadcastWhenLowerRank(self):
      a = [[1.0, 2], [2.0, 3]]
      b = [[1.0, 2], [2.0, 3]]
      x = [0.5, 0.5]
      pdf = beta_lib.Beta(a, b).prob(x)
      self.assertAllClose(
          [[1.0, 3.0 / 2], [3.0 / 2, 15.0 / 8]],
          self.evaluate(pdf),
          rtol=1e-5,
          atol=1e-5,
      )
      self.assertEqual((2, 2), pdf.get_shape())
  
imports: import importlib
  import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import random_seed
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops.distributions import beta
  from tensorflow.python.ops.distributions import kullback_leibler
  from tensorflow.python.platform import test
  from tensorflow.python.platform import tf_logging
label_desc: Use image node_id 13 for calling the BetaTest obj's underlying member method code with example usage: obj.testPdfXStretchedInBroadcastWhenLowerRank() without return types