Thanks for reply.
Build error happened with the code you mentioned so I modified it like below,
int out_ndim = 2;
int64_t out_shape[2] = {1，1000, };
But the result says,
daeinki@daeinki-linux:~/project/s5pc210/public/working/tvm_pc/tvm/build/test/app_test_pc$ ./mxnet_resnet
param size : 46765317
The maximum position in output vector is: 0
0 is correct? How could I check the label value with 0?
I guess below code needs to be modified correctly because out_ndim and out_shape are changed.
66 // get the function from the module(get output data)
67 tvm::runtime::PackedFunc get_output = mod.GetFunction(“get_output”);
68 get_output(0, y);
69
70 // get the maximum position in output vector
71 auto y_iter = static_cast<float*>(y->data);
72 auto max_iter = std::max_element(y_iter, y_iter + 1000);
73 auto max_index = std::distance(y_iter, max_iter);
74 std::cout << "The maximum position in output vector is: " << max_index << std::endl;
Thanks,
Inki Dae