def METHOD_NAME(self):
    target_classes = np.array([[1, 2, 3, 4]])
    target_classes = ops.expand_dims(target_classes, axis=-1)
    indices = np.array([[0, 2]], dtype="int32")
    expected_classes = np.array([[1, 3]])
    expected_classes = ops.expand_dims(expected_classes, axis=-1)
    res = _target_gather(target_classes, indices)
    self.assertAllClose(expected_classes, res)