OnnxSetOutputShape
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);
}