code: def test_gather_nonscalar_elements(self):
      ta = nptf.compat.v2.TensorArray(
          dtype=np.int32, tensor_array_name="foo", size=3
      ).unstack(np.int32([[0, 0], [1, 1], [2, 2]]))
      self.assertAllEqual([1, 1], ta.gather([1])[0])
  
file_path: probability/tensorflow_probability/python/internal/backend/numpy/tensor_array_ops_test.py
imports: import numpy
  from tensorflow_probability.python.internal.backend import numpy
  from tensorflow_probability.substrates.numpy.internal import test_util
label_desc: Use image node_id 4 for calling the TensorArrayTest obj's underlying member method code with example usage: obj.test_gather_nonscalar_elements() without return types