code: def test_params_rank3_indices_rank1_axis_1(self):
      axis = 1
      params = np.random.randint(10, 100, size=[4, 5, 2])
      indices = np.random.randint(0, params.shape[axis], size=[6])
  
      result = util.index_remapping_gather(params, indices, axis=axis)
      self.assertAllEqual(
          params.shape[:axis]
          + indices.shape[:1]
          + params.shape[axis + 1 :],
          result.shape,
      )
      result_ = self.evaluate(result)
  
      for i in range(params.shape[0]):
          for j in range(indices.shape[0]):
              for k in range(params.shape[2]):
                  self.assertEqual(
                      params[i, indices[j], k], result_[i, j, k]
                  )
  
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 5 for calling the IndexRemappingGatherTest obj's underlying member method code with example usage: obj.test_params_rank3_indices_rank1_axis_1() without return types