code: def test_pick_partial_shape():
      x = mx.sym.Variable("x")
      index = mx.sym.Variable("index")
      y = mx.sym.pick(x, index, axis=1)
      # batch size unknown
      _, result, _ = y.infer_shape_partial(
          x=(0, 3, 3),
          index=(
              0,
              3,
          ),
      )
      assert result == [(0, 3)]
      with mx.np_shape(True):
          _, result, _ = y.infer_shape_partial(
              x=(-1, 3, 3),
              index=(
                  -1,
                  3,
              ),
          )
          assert result == [(-1, 3)]
  
file_path: mxnet/tests/python/unittest/test_infer_shape.py
imports: import mxnet
  from common import models
  import pytest
label_desc: Use image node_id 15 for calling a global function with example usage: test_pick_partial_shape() without return types