函数:get_input_size_by_index
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
根据aclmdlDesc类型的数据,获取指定输入的大小,单位为Byte。
函数原型
- C函数原型
1
size_t aclmdlGetInputSizeByIndex(aclmdlDesc *modelDesc, size_t index)
- python函数
1
size = acl.mdl.get_input_size_by_index(model_desc, index)
参数说明
参数名 |
说明 |
---|---|
model_desc |
int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 |
index |
int,指定获取第几个输入的大小,“index”取值从0开始。 |
返回值说明
返回值 |
说明 |
---|---|
size |
int,返回指定输入的大小,单位是Byte。
|
约束说明
如果模型输入的Shape是动态的、且维度的取值为-1(表示此维度可以使用大于或等于1的任意取值),则通过本接口获取的大小为0,用户需根据实际数据占用的内存大小来申请内存。
父主题: aclmdlDesc