code: def testBijectorToTransformFnMulti(self):
      bijector = DupBijector()
      state = jnp.ones([1, 2], dtype=self._dtype)
      transform_fn = util_tfp.bijector_to_transform_fn(
          bijector, state_structure=state, batch_ndims=1
      )
  
      fwd, (_, fwd_ldj1), fwd_ldj2 = fun_mc.call_transport_map_with_ldj(
          transform_fn, state
      )
      self.assertAllClose([np.ones([1, 2]), np.ones([1, 2])], fwd)
  
      self.assertAllClose(0.0, fwd_ldj1)
      self.assertAllClose(0.0, fwd_ldj2)
  
      inverse_transform_fn = util.inverse_fn(transform_fn)
      inv, (_, inv_ldj1), inv_ldj2 = fun_mc.call_transport_map_with_ldj(
          inverse_transform_fn,
          [
              jnp.ones([1, 2], dtype=self._dtype),
              jnp.ones([2, 1], dtype=self._dtype),
          ],
      )
      self.assertAllClose(np.ones([1, 2]), inv)
      self.assertAllClose(0.0, inv_ldj1)
      self.assertAllClose(0.0, inv_ldj2)
  
imports: from absl.testing import parameterized
  from jax import config
  import numpy
  import tensorflow.compat.v2
  from fun_mc import backend
  from fun_mc import fun_mc_lib
  from fun_mc import test_util
  from fun_mc import util_tfp
label_desc: Use image node_id 5 for calling the UtilTFPTest obj's underlying member method code with example usage: obj.testBijectorToTransformFnMulti() without return types