code: def test_matmat(self, left_shape, right_shape, result_shape):
      matmat = jnp.vectorize(jnp.dot, signature="(n,m),(m,k)->(n,k)")
      self.assertEqual(
          matmat(jnp.zeros(left_shape), jnp.zeros(right_shape)).shape,
          result_shape,
      )
  
imports: from functools import partial
  from absl.testing import absltest
  import numpy
  import jax
  from jax import numpy
  from jax._src import test_util
  from jax import config
label_desc: Use image node_id 1 for calling the VectorizeTest obj's underlying member method code with example usage: obj.test_matmat(left_shape, right_shape, result_shape) without return types