code: def test_static_shape_using_placeholder_for_dim(self):
      @def_function.function(
          input_signature=[
              tensor_spec.TensorSpec(shape=[], dtype=dtypes.int32)
          ]
      )
      def f(dim):
          s = math_ops.sobol_sample(dim, 100, dtype=dtypes.float32)
          assert s.shape.as_list() == [100, None]
          return s
  
      self.assertAllEqual([100, 10], self.evaluate(f(10)).shape)
  
imports: import numpy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import gen_math_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 5 for calling the SobolSampleOpTest obj's underlying member method code with example usage: obj.test_static_shape_using_placeholder_for_dim() and returns: s