code: def test_rank_2_and_axis_0(self):
      params = [[95, 46, 17], [46, 29, 55]]
      indices = [[0, 0, 1], [1, 0, 1]]
  
      expected = [[95, 46, 55], [46, 46, 55]]
      result = util.index_remapping_gather(params, indices)
  
      self.assertAllEqual(np.asarray(params).shape, result.shape)
  
      self.assertAllEqual(expected, self.evaluate(result))
  
imports: import collections
  import warnings
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import lognormal
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.mcmc.internal import util
label_desc: Use image node_id 2 for calling the IndexRemappingGatherTest obj's underlying member method code with example usage: obj.test_rank_2_and_axis_0() without return types