code: def test_krprod_two_inputs():
      A = mx.nd.arange(1, 7).reshape((3, 2))
      B = mx.nd.arange(1, 3).reshape((1, 2))
      out = mx.nd.khatri_rao(A, B)
      expected = mx.nd.array([[1, 4], [3, 8], [5, 12]])
      assert_mx_allclose(out, expected, rtol=1e-12)
  
      A = mx.nd.arange(1, 7).reshape((3, 2))
      B = mx.nd.arange(1, 9).reshape((4, 2))
      out = mx.nd.khatri_rao(A, B)
      expected = mx.nd.array(
          [
              [1, 4],
              [3, 8],
              [5, 12],
              [7, 16],
              [3, 8],
              [9, 16],
              [15, 24],
              [21, 32],
              [5, 12],
              [15, 24],
              [25, 36],
              [35, 48],
          ]
      )
      assert_mx_allclose(out, expected, 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 3 for calling a global function with example usage: test_krprod_two_inputs() without return types