10 #include <onnxruntime_cxx_api.h>
23 for (
const auto& feature : features) total_size += feature.size();
25 std::vector<T> flatten1D;
26 flatten1D.reserve(total_size);
28 for (
const auto& feature : features)
29 for (
const auto& elem : feature)
30 flatten1D.push_back(elem);
42 std::vector<std::vector<int64_t> >& dataShape,
43 std::vector<std::string>& nodeNames);
52 std::vector<std::vector<int64_t> >& dataShape,
53 std::vector<std::string>& nodeNames);
58 std::vector<std::vector<int64_t> >& dataShape,
59 std::vector<std::string>& nodeNames,
70 const std::vector<std::string>& inputNames,
71 const std::vector<Ort::Value>& inputData,
73 const std::vector<Ort::Value>& outputData
80 Ort::MemoryInfo memoryInfo = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
81 return Ort::Value::CreateTensor<T>(memoryInfo,
data.data(),
data.size(), dataShape.data(), dataShape.size());