docstr: Test for batch seeds.
code: def testBatchSeeds(self, alg):
      """Test for batch seeds."""
      shape = [2, 3]
      count = 6
      gen = random.Generator.from_seed(1234, alg=alg)
      if alg == random_ops_util.Algorithm.THREEFRY:
          # We don't have CPU/GPU kernels for ThreeFry yet.
          return
      keys1 = gen._make_int64_keys(shape=shape)
      keys2 = gen._make_int64_keys(shape=shape)
      self.assertAllDifferent([keys1, keys2])
      seeds1 = gen.make_seeds(count=count)
      seeds2 = gen.make_seeds(count=count)
      self.assertAllDifferent([seeds1[0, :], seeds2[0, :]])
      gens = gen.split(count=count)
      self.assertAllEqual(count, len(gens))
      randoms = [
          g.uniform_full_int(shape=shape, dtype=dtypes.int32)
          for g in gens
      ]
      self.assertAllDifferent(randoms)
  
      # Tests graph mode.
      @def_function.function
      def f():
          return gen.make_seeds(count=count)
  
      for _ in range(3):
          f()
  
imports: import os
  import re
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.kernel_tests.random import util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_random_ops
  from tensorflow.python.ops import gen_stateful_random_ops
  from tensorflow.python.ops import logging_ops
  from tensorflow.python.ops import random_ops_util
  from tensorflow.python.ops import stateful_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the StatefulRandomOpsTest obj's underlying member method code with example usage: obj.testBatchSeeds(alg) and returns: gen