class_docstr: Tests the correctness of the CorrelationCholesky bijector.
class_docstr_tok: ["Tests","the","correctness","of","the","CorrelationCholesky","bijector","."]
code: def testShape(self):
      x_shape = tf.TensorShape([5, 4, 6])
      y_shape = tf.TensorShape([5, 4, 4, 4])
  
      b = correlation_cholesky.CorrelationCholesky(validate_args=True)
  
      x = tf.ones(shape=x_shape, dtype=tf.float32)
      y_ = b.forward(x)
      self.assertAllEqual(
          tensorshape_util.as_list(y_.shape),
          tensorshape_util.as_list(y_shape),
      )
      x_ = b.inverse(y_)
      self.assertAllEqual(
          tensorshape_util.as_list(x_.shape),
          tensorshape_util.as_list(x_shape),
      )
  
      y_shape_ = b.forward_event_shape(x_shape)
      self.assertAllEqual(
          tensorshape_util.as_list(y_shape_),
          tensorshape_util.as_list(y_shape),
      )
      x_shape_ = b.inverse_event_shape(y_shape)
      self.assertAllEqual(
          tensorshape_util.as_list(x_shape_),
          tensorshape_util.as_list(x_shape),
      )
  
      y_shape_tensor = self.evaluate(
          b.forward_event_shape_tensor(
              tensorshape_util.as_list(x_shape)
          )
      )
      self.assertAllEqual(
          y_shape_tensor, tensorshape_util.as_list(y_shape)
      )
      x_shape_tensor = self.evaluate(
          b.inverse_event_shape_tensor(
              tensorshape_util.as_list(y_shape)
          )
      )
      self.assertAllEqual(
          x_shape_tensor, tensorshape_util.as_list(x_shape)
      )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.bijectors import correlation_cholesky
  from tensorflow_probability.python.bijectors import fill_triangular
  from tensorflow_probability.python.distributions import beta
  from tensorflow_probability.python.distributions import cholesky_lkj
  from tensorflow_probability.python.distributions import lkj
  from tensorflow_probability.python.distributions.internal import statistical_testing
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc import hmc
  from tensorflow_probability.python.mcmc import sample
  from tensorflow_probability.python.mcmc import transformed_kernel
label_desc: Use image node_id 3 for calling the CorrelationCholeskyBijectorTest obj's underlying member method code with example usage: obj.testShape() without return types