code: def test_batch_dot_partial_shape():
      x = mx.sym.Variable("x")
      y = mx.sym.Variable("y")
      z = mx.sym.batch_dot(x, y)
      # lhs and rhs batch size unknown
      _, result_shape, _ = z.infer_shape_partial(
          x=(0, 3, 4), y=(0, 4, 5)
      )
      assert result_shape == [(0, 3, 5)]
      # rhs second dim unknown
      _, result_shape, _ = z.infer_shape_partial(
          x=(0, 3, 4), y=(0, 0, 5)
      )
      assert result_shape == [()]
      with mx.np_shape(True):
          _, result_shape, _ = z.infer_shape_partial(
              x=(-1, 3, 4), y=(-1, 4, 5)
          )
          assert result_shape == [(-1, 3, 5)]
          _, result_shape, _ = z.infer_shape_partial(
              x=(-1, 3, 4), y=(-1, -1, 5)
          )
          assert result_shape == [None]
  
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 12 for calling a global function with example usage: test_batch_dot_partial_shape() without return types