Привязка ввода-вывода ONNX ⇐ C++

[Anonymous]

Мне нужно связать тензорный ввод и вывод с помощью привязки ввода-вывода для модели времени выполнения ONNX. но я не получил вывода. выходной тензор возвращает NULL-указатель. Я приложу код ниже.

Код: Выделить всё

std::vector input_tensors;
std::vector output_tensors;
std::vector input_node_names_c_str;
std::vector output_node_names_c_str;
int64_t input_height = input_node_dims[0].at(2);
int64_t input_width = input_node_dims[0].at(3);

// // Pass gpu_graph_id to RunOptions through RunConfigs
Ort::RunOptions run_option;
// gpu_graph_id is optional if the session uses only one cuda graph
run_option.AddConfigEntry("gpu_graph_id", "1");

// Dimension expansion [CHW -> NCHW]
std::vector input_tensor_shape = {1, 3, input_height, input_width};
std::vector output_tensor_shape = {1, 300, 84};
size_t input_tensor_size = vector_product(input_tensor_shape);
size_t output_tensor_size = vector_product(output_tensor_shape);
std::vector input_tensor_values(p_blob, p_blob + input_tensor_size);

Ort::IoBinding io_binding{session};
Ort::MemoryInfo memory_info = Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeCPU);

input_tensors.push_back(Ort::Value::CreateTensor(
memory_info, input_tensor_values.data(), input_tensor_size,
input_tensor_shape.data(), input_tensor_shape.size()
));

// Check if input and output node names are empty
for (const auto& inputNodeName : input_node_names) {
if (std::string(inputNodeName).empty()) {
std::cerr 

Подробнее здесь: [url]https://stackoverflow.com/questions/78506207/onnx-i-o-binding[/url]