code: def test_transform(self):
      seed = jax.random.PRNGKey(20231018)
      bnn = kernels.LinearBNN(width=5)
      bnn.likelihood_model.noise_min = 0.2
      transform, _, _ = util.make_transforms(bnn)
      p = bnn.init(seed, jnp.ones((1, 10), dtype=jnp.float32))
  
      # Softplus(low=0.2) bijector
      self.assertEqual(
          0.2 + jax.nn.softplus(p["params"]["noise_scale"]),
          transform(p)["params"]["noise_scale"],
      )
      self.assertEqual(
          jnp.exp(p["params"]["amplitude"]),
          transform(p)["params"]["amplitude"],
      )
  
      # Identity bijector
      self.assertAllEqual(
          p["params"]["dense2"]["kernel"],
          transform(p)["params"]["dense2"]["kernel"],
      )
  
file_path: probability/tensorflow_probability/python/experimental/autobnn/util_test.py
imports: import jax
  import jax.numpy
  import numpy
  from tensorflow_probability.python.experimental.autobnn import kernels
  from tensorflow_probability.python.experimental.autobnn import util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 2 for calling the UtilTest obj's underlying member method code with example usage: obj.test_transform() without return types