code: def test_krprod_one_input():
      A = mx.nd.arange(1, 9).reshape((2, 4))
      out = mx.nd.khatri_rao(A)
      assert_mx_allclose(out, A, rtol=1e-12)
  
imports: from __future__ import print_function
  import numpy
  import mxnet
  from numpy.testing import assert_allclose
label_desc: Use image node_id 2 for calling a global function with example usage: test_krprod_one_input() without return types