C函数原型 |
aclError aclmdlGetInputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims) |
---|---|
Python函数 |
dims, ret = acl.mdl.get_input_dims(model_desc, index) |
函数功能 |
根据模型描述获取模型的输入tensor的维度信息。同步接口。 如果模型中含有静态AIPP配置信息,您可以根据实际需要选择acl.mdl.get_input_dims接口或acl.mdl.get.input_dims_v2接口查询维度信息,两者的区别在于: |
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针。 index:int,指定获取第几个输入的dims,index值从0开始。 |
返回值说明 |
dims:dict,针对多batch或多种宽高的场景,按照该字典的数组返回其支持的最大档的组合
ret:int,错误码。
|
约束说明 |
无 |
注意事项 |
无 |