函数:get_output_size_by_index
C函数原型 |
size_t aclmdlGetOutputSizeByIndex(aclmdlDesc *modelDesc, size_t index) |
---|---|
Python函数 |
size = acl.mdl.get_output_size_by_index(model_desc, index) |
函数功能 |
根据aclmdlDesc类型的数据,获取指定输出的大小,单位为Byte。同步接口。 |
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针。需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输出的大小,index值从0开始。 |
返回值说明 |
size:int,返回指定输出的大小,单位是Byte。 针对动态Batch、动态分辨率(宽高)的场景,返回最大档位对应的输出的大小。静态场景下,返回指定输出的大小。单位是Byte。 |
约束说明 |
如果通过本接口获取的大小为0,有可能是由于输出Shape的范围不确定,用户需先根据实际情况预估一块较大的输出内存,在模型执行过程中,系统会校验用户指定的输出内存大小是否符合要求,如果不符合要求,系统会返回报错,并在报错信息中提示具体需要多大的输出内存。您可以通过以下两种方式查看报错:
|
注意事项 |
无 |
父主题: aclmdlDesc