OnnxSetInputShape

通过索引设置模型输入数据的形状。

bool  OnnxSetInputShape(

   long          onnx_handle,  // ONNX会话句柄

   long          input_index,  // 输入参数索引

   const ulong&  shape[]       // 描述输入数据形状的数组

   );

参数

onnx_handle

[in]  通过OnnxCreate或OnnxCreateFromBuffer创建的ONNX会话对象句柄。

input_index

[in]  输入参数的索引，从0开始。

shape

[in]  描述模型输入数据形状的数组。

返回值

如果成功，返回输入参数的名称；否则返回NULL。要获取错误代码，请调用GetLastError函数。

例如：

//---- 描述模型输入和输出数据的形状

   const long  ExtOutputShape[] = {1,1};

   const long  ExtInputShape [] = {1,10,4};

//--- 创建模型

   long handle=OnnxCreateFromBuffer(model,ONNX_DEBUG_LOGS);

//--- 指定输入数据形状

   if(!OnnxSetInputShape(handle,0,ExtInputShape))

     {

      Print("failed, OnnxSetInputShape error ",GetLastError());

      OnnxRelease(handle);

      return(-1);

     }

//--- 指定输出数据形状

   if(!OnnxSetOutputShape(handle,0,ExtOutputShape))

     {

      Print("failed, OnnxSetOutputShape error ",GetLastError());

      OnnxRelease(handle);

      return(-1);

     }

另见

OnnxSetOutputShape