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