code: def test_unknown_shape(self):
      @def_function.function
      def shuffle(repeats):
          indices = array_ops.repeat(2, repeats)
          return stateless.index_shuffle(
              indices, seed=(1, 2), max_index=10
          )
  
      new_index = shuffle(constant_op.constant(2))
      self.assertAllGreaterEqual(new_index, 0)
      self.assertAllLessEqual(new_index, 10)
  
file_path: tensorflow/tensorflow/python/kernel_tests/random/random_index_shuffle_test.py
imports: import itertools
  from absl.testing import parameterized
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_random_index_shuffle_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 4 for calling the StatelessOpsTest obj's underlying member method code with example usage: obj.test_unknown_shape() and returns: stateless