更新时间:2021-03-18 GMT+08:00
分享

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表示失败。

分享:

    相关文档

    相关产品

关闭导读