量化交易机器人系统开发详细策略/需求步骤/逻辑方案/源码设计

// wrapping input tensor, convert nhwc to nchw
std::vector dims{1, INPUT_SIZE, INPUT_SIZE, 3};
auto nhwc_Tensor = MNN::Tensor::create(dims, NULL, MNN::Tensor::TENSORFLOW);
auto nhwc_data = nhwc_Tensor->host();
auto nhwc_size = nhwc_Tensor->size();
::memcpy(nhwc_data, image.data, nhwc_size);

std::string input_tensor = "data";
auto inputTensor  = net->getSessionInput(session, nullptr);
inputTensor->copyFromHostTensor(nhwc_Tensor);

// run network
net->runSession(session);

// get output data
std::string output_tensor_name0 = "conv5_fwd";

MNN::Tensor *tensor_lmks  = net->getSessionOutput(session, output_tensor_name0.c_str());

MNN::Tensor tensor_lmks_host(tensor_lmks, tensor_lmks->getDimensionType());

tensor_lmks->copyToHostTensor(&tensor_lmks_host);