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