code: def _matmul(a, b):
      xp = backend.get_array_module(a)
      if not hasattr(xp, "matmul"):
          # NumPy 1.9 does not support matmul. We use einsum instead.
          return xp.einsum("ijl,ilk->ijk", a, b)
      return xp.matmul(a, b)
  
imports: import numpy
  import chainer
  from chainer import backend
  from chainer.backends import cuda
  from chainer.backends import intel64
  from chainer import configuration
  from chainer import function_node
  import chainer.functions
  from chainer import memory_layouts
  from chainer.utils import argument
  from chainer.utils import conv
  from chainer.utils import type_check
  import chainerx
label_desc: Use image node_id 2 for calling a global function with example usage: _matmul(a, b) and returns: xp, xp