code: def test_embedding_partial_shape():
      # testing embedding with batch size unknown
      x = mx.sym.Variable("x")
      w = mx.sym.Variable("w")
      y = mx.sym.Embedding(
          data=x, weight=w, input_dim=100, output_dim=10
      )
      _, result_shape, _ = y.infer_shape_partial(x=(0, 5), w=(100, 10))
      assert result_shape == [(0, 5, 10)]
      with mx.np_shape(True):
          _, result_shape, _ = y.infer_shape_partial(
              x=(-1, 5), w=(100, 10)
          )
          assert result_shape == [(-1, 5, 10)]
  
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 13 for calling a global function with example usage: test_embedding_partial_shape() without return types