code: def testRandomSeeds(self):
      def _loop_fn(state, _):
          return samplers.split_seed(state, 2)[1]
  
      def _trace_fn(state):
          return [
              state,
              samplers.split_seed(state, ps.convert_to_shape_tensor(5)),
              tf.broadcast_to(
                  state, tf.TensorShape([2, 4]) + state.shape
              ),
          ]
  
      _, trace = loop_util.trace_scan(
          loop_fn=_loop_fn,
          initial_state=test_util.test_seed(sampler_type="stateless"),
          elems=[1, 2, 3],
          trace_fn=_trace_fn,
      )
  
      self.assertAllEqual(3, trace[0].shape[0])
      self.assertAllEqual((3, 5), trace[1].shape[:2])
      self.assertAllEqual((3, 2, 4), trace[2].shape[:3])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.internal import auto_composite_tensor
  from tensorflow_probability.python.internal import loop_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.internal import samplers
  from tensorflow_probability.python.internal import tensorshape_util
  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 TraceScanTest obj's underlying member method code with example usage: obj.testRandomSeeds() and returns: samplers, list