- Suporte a ONNX
- Conversão de formatos
- Autoconversão de dados
- Criação de modelo
- Inicialização do modelo
- Verificação no testador
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- Estruturas de dados
OnnxSetInputShape
Define a dimensionalidade dos dados de entrada do modelo por índice.
bool OnnxSetInputShape(
|
Parâmetros
onnx_handle
[in] Identificador do objeto da sessão ONNX, criado por meio de OnnxCreate ou OnnxCreateFromBuffer.
input_index
[in] Índice do parâmetro de entrada, a partir de 0.
shape
[in] Matriz que descreve a dimensionalidade dos dados de entrada.
Valor retornado
Retorna o nome do parâmetro de entrada se for bem-sucedido; caso contrário, retorna NULL. Para obter o código de erro chame a função GetLastError.
Exemplo:
//---- descrevemos a dimensionalidade dos dados de entrada do modelo
|
Veja também