docstr: Different `Axis`-th (non-0) dimension: shape [1, 1], [1, 2] -> [1, 3].
code: def testGatherDiffShapeAtAxis1(self, strategy, pure_eager):
      """Different `Axis`-th (non-0) dimension: shape [1, 1], [1, 2] -> [1, 3]."""
  
      def value_fn(ctx):
          return constant_op.constant(
              1, shape=(1, ctx.replica_id_in_sync_group + 1)
          )
  
      distributed_values = (
          strategy.experimental_distribute_values_from_function(
              value_fn
          )
      )
      axis = 1
  
      def run():
          return strategy.gather(distributed_values, axis=axis)
  
      if not pure_eager:
          run = def_function.function(run)
  
      expected_result = constant_op.constant(
          1, shape=(1, sum(range(strategy.num_replicas_in_sync + 1)))
      )
  
      self.assertAllEqual(expected_result, run().numpy())
  
imports: from absl.testing import parameterized
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.distribute import central_storage_strategy
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import mirrored_strategy
  from tensorflow.python.distribute import strategy_combinations
  from tensorflow.python.distribute import test_util
  from tensorflow.python.distribute import tpu_strategy
  from tensorflow.python.distribute.collective_all_reduce_strategy import CollectiveAllReduceStrategy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import indexed_slices
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.platform import test
  from tensorflow.python.util import nest
label_desc: Use image node_id 9 for calling the GatherTest obj's underlying member method code with example usage: obj.testGatherDiffShapeAtAxis1(strategy, pure_eager) and returns: constant_op, strategy