- Поддержка ONNX
- Преобразование форматов
- Автоконвертация данных
- Cоздание модели
- Запуск модели
- Проверка в тестере
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- Структуры данных
OnnxSetInputShape
Устанавливает размерность входных данных модели по индексу.
bool OnnxSetInputShape(
|
Параметры
onnx_handle
[in] Хэндл объекта сессии ONNX, созданный через OnnxCreate или OnnxCreateFromBuffer.
input_index
[in] Индекс входного параметра, начиная с 0.
shape
[in] Массив, описывающий размерность входных данных модели.
Возвращаемое значение
В случае успешного выполнения возвращает имя входного параметра, иначе NULL. Для получения кода ошибки вызовите функцию GetLastError.
Пример:
//---- опишем размерности входных и выходных данных модели
|
Смотри также