OnnxSetOutputShape

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

bool  OnnxSetOutputShape(

   long          onnx_handle,   // ONNX会话句柄

   long          output_index,  // 输出参数索引

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

   );

参数

onnx_handle

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

output_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);

     }

另见

OnnxSetInputShape