aclmdlGetOutputDims
函数功能
根据模型描述获取指定的模型输出tensor的维度信息。
固定Shape场景下,通过该接口获取指定的模型输出tensor的维度信息。
动态Shape(动态Batch或动态分辨率)场景下,通过该接口获取最大档的维度信息。
函数原型
aclError aclmdlGetOutputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
modelDesc |
输入 |
aclmdlDesc类型数据的指针。 |
index |
输入 |
指定获取第几个输出的Dims,index值从0开始。 |
dims |
输出 |
获取输出维度信息。 |
返回值说明
返回0表示成功,返回非0表示失败。
