code: def testScalar2D(self):
      with self.session() as session, self.test_scope():
          data = np.array(
              [
                  [0, 1, 2],
                  [3, 4, 5],
                  [6, 7, 8],
                  [9, 10, 11],
                  [12, 13, 14],
              ]
          )
          for dtype in self.all_tf_types:
              for axis in 0, 1, -1:
                  params_np = self._buildParams(data, dtype)
                  params = array_ops.placeholder(dtype=dtype)
                  indices = constant_op.constant(2)
                  gather_t = array_ops.gather(
                      params, indices, axis=axis
                  )
                  gather_val = session.run(
                      gather_t, feed_dict={params: params_np}
                  )
                  expected = constant_op.constant(
                      np.take(params_np, 2, axis=axis), dtype
                  )
                  self.assertAllEqual(expected, gather_val)
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  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 variables
  from tensorflow.python.platform import flags
  from tensorflow.python.platform import test
label_desc: Use image node_id 3 for calling the GatherTest obj's underlying member method code with example usage: obj.testScalar2D() without return types