@@ -167,7 +167,7 @@ std::vector<O> OrtModel::inference(std::vector<I>& input)
167167{
168168 std::vector<int64_t > inputShape{(int64_t )(input.size () / mInputShapes [0 ][1 ]), (int64_t )mInputShapes [0 ][1 ]};
169169 std::vector<Ort::Value> inputTensor;
170- inputTensor.emplace_back (Ort::Value::CreateTensor<O>(pImplOrt->memoryInfo , ( reinterpret_cast <O*>(input)) .data (), input.size (), inputShape.data (), inputShape.size ()));
170+ inputTensor.emplace_back (Ort::Value::CreateTensor<O>(pImplOrt->memoryInfo , reinterpret_cast <O*>(input.data () ), input.size (), inputShape.data (), inputShape.size ()));
171171 // input.clear();
172172 auto outputTensors = (pImplOrt->session )->Run (pImplOrt->runOptions , inputNamesChar.data (), inputTensor.data (), inputTensor.size (), outputNamesChar.data (), outputNamesChar.size ());
173173 O* outputValues = reinterpret_cast <O*>(outputTensors[0 ].template GetTensorMutableData <O>());
@@ -182,7 +182,7 @@ std::vector<O> OrtModel::inference(std::vector<std::vector<I>>& input)
182182 std::vector<Ort::Value> inputTensor;
183183 for (auto i : input) {
184184 std::vector<int64_t > inputShape{(int64_t )(i.size () / mInputShapes [0 ][1 ]), (int64_t )mInputShapes [0 ][1 ]};
185- inputTensor.emplace_back (Ort::Value::CreateTensor<O>(pImplOrt->memoryInfo , ( reinterpret_cast <O*>(i)) .data (), i.size (), inputShape.data (), inputShape.size ()));
185+ inputTensor.emplace_back (Ort::Value::CreateTensor<O>(pImplOrt->memoryInfo , reinterpret_cast <O*>(i.data () ), i.size (), inputShape.data (), inputShape.size ()));
186186 }
187187 // input.clear();
188188 auto outputTensors = (pImplOrt->session )->Run (pImplOrt->runOptions , inputNamesChar.data (), inputTensor.data (), inputTensor.size (), outputNamesChar.data (), outputNamesChar.size ());
0 commit comments