code: def test_debatchify_params1(self):
      out = training_util.debatchify_params(
          {"noise_scale": jnp.ones(10)}
      )
      self.assertLen(out, 10)
      self.assertEqual(out[0], {"noise_scale": jnp.ones(1)})
      self.assertEqual(out[1], {"noise_scale": jnp.ones(1)})
  
imports: import chex
  import jax
  import jax.numpy
  import numpy
  from tensorflow_probability.python.experimental.autobnn import kernels
  from tensorflow_probability.python.experimental.autobnn import operators
  from tensorflow_probability.python.experimental.autobnn import training_util
  from tensorflow_probability.python.experimental.autobnn import util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 6 for calling the TrainingUtilTest obj's underlying member method code with example usage: obj.test_debatchify_params1() without return types