code: def testLogPDFMultidimensional(self):
      batch_size = 6
      a = tf.constant(1.0)
      b = tf.constant([[1.0, 2.0]] * batch_size)
      x = np.array([[0.0, 0.1, 0.2, 0.4, 0.5, 1.0]], dtype=np.float32).T
      dist = sigmoid_beta.SigmoidBeta(a, b, validate_args=True)
      oracle = transformed_distribution.TransformedDistribution(
          distribution=beta.Beta(a, b),
          bijector=invert.Invert(sigmoid.Sigmoid()),
      )
      log_pdf = dist.log_prob(x)
      expected_logpdf = oracle.log_prob(x)
      self.assertEqual(log_pdf.shape, (6, 2))
      self.assertAllClose(log_pdf, expected_logpdf)
      pdf = dist.prob(x)
      expected_pdf = oracle.prob(x)
      self.assertEqual(pdf.shape, (6, 2))
      self.assertAllClose(pdf, expected_pdf)
  
imports: import numpy
  from scipy import stats
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import invert
  from tensorflow_probability.python.bijectors import sigmoid
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import sigmoid_beta
  from tensorflow_probability.python.distributions import transformed_distribution
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.internal import assert_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 11 for calling the SigmoidBetaTest obj's underlying member method code with example usage: obj.testLogPDFMultidimensional() without return types