code: def testBijectorWithoutActivation(self):
      width = 4
      x = samplers.uniform(
          (2, width, width),
          minval=-1.0,
          maxval=1.0,
          seed=test_util.test_seed(sampler_type="stateless"),
      )
  
      bijector = highway_flow.build_trainable_highway_flow(
          width, activation_fn=False, seed=test_util.test_seed()
      )
      self.evaluate(
          [v.initializer for v in bijector.trainable_variables]
      )
      self.assertStartsWith(bijector.name, "highway_flow")
      self.assertAllClose(
          x, bijector.inverse(tf.identity(bijector.forward(x)))
      )
      self.assertAllClose(
          bijector.forward_log_det_jacobian(x, event_ndims=2),
          -bijector.inverse_log_det_jacobian(
              tf.identity(bijector.forward(x)), event_ndims=2
          ),
      )
  
file_path: probability/tensorflow_probability/python/experimental/bijectors/highway_flow_test.py
imports: from absl.testing import parameterized
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.experimental.bijectors import highway_flow
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 2 for calling the HighwayFlowTests obj's underlying member method code with example usage: obj.testBijectorWithoutActivation() without return types