code: def testWorksWithChain(self):
      shape_out = (4,)
      shape_in = (2, 2)
      x = np.zeros(shape_in)
      y = np.zeros(shape_out)
      bijector = chain.Chain(
          [
              identity.Identity(),
              reshape.Reshape(
                  event_shape_out=shape_out, event_shape_in=shape_in
              ),
          ]
      )
      new_y = self.evaluate(bijector.forward(x))
      new_x = self.evaluate(bijector.inverse(y))
      fldj = self.evaluate(
          bijector.forward_log_det_jacobian(
              x, event_ndims=len(shape_in)
          )
      )
      ildj = self.evaluate(
          bijector.inverse_log_det_jacobian(
              y, event_ndims=len(shape_out)
          )
      )
      self.assertEqual(shape_out, new_y.shape)
      self.assertEqual(shape_in, new_x.shape)
      self.assertEqual((), fldj.shape)
      self.assertEqual((), ildj.shape)
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.bijectors import chain
  from tensorflow_probability.python.bijectors import identity
  from tensorflow_probability.python.bijectors import reshape
  from tensorflow_probability.python.internal import hypothesis_testlib
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 5 for calling the ReshapeBijectorTestStatic obj's underlying member method code with example usage: obj.testWorksWithChain() without return types