code: def testSimpleGather(self, indices_dtype):
      data = np.array([0, 1, 2, 3, 7, 5, 8, 9, 10, 11, 15, 13])
      indices = [3, 4]
      with self.session():
          for dtype in _TEST_TYPES:
              params_np = self._buildParams(data, dtype)
              params = constant_op.constant(params_np)
              indices_tf = constant_op.constant(
                  indices, dtype=indices_dtype
              )
              gather_t = array_ops.batch_gather(params, indices_tf)
              expected_result = np.array([3, 7])
              np_val = self._buildParams(expected_result, dtype)
              gather_val = self.evaluate(gather_t)
              self.assertAllEqual(np_val, gather_val)
              self.assertEqual(np_val.shape, gather_t.get_shape())
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/array_ops/batch_gather_op_test.py
imports: from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling the GatherTest obj's underlying member method code with example usage: obj.testSimpleGather(indices_dtype) without return types