How to use run_from_inputs method in Java

#1

I use run_from_inputs like this:

public GraphModule runFromInput(int[] index, NDArray[] value) {
    for (int i = 0; i < index.length; i++) {
      fRunFromInputs.pushArg(index[i]).pushArg(value[i]);
    }
    fRunFromInputs.invoke();
    return this;
}

but get a error: Check failed: type_code_ == kDLInt (13 vs. 0) : expected int but got NDArrayContainer Stack trace: File “/opt/meituan/git/banma_algorithm_tvm/include/tvm/runtime/packed_func.h”, line 562 0: tvm::runtime::TVMPODValue_::operator int() const

the source code of run_from_inputs is:

image
image1628×886 114 KB

How should I write the correct Java code? thank you all