函数:get_input_dims
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类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输入的dims,index值从0开始。  | 
返回值说明  | 
dims:dict,针对多batch或多种宽高的场景,按照该字典的数组返回其支持的最大档的组合,具体请参见aclmdlIODims。 dims = {
        "name": xxx, #tensor name
        "dimCount":xxx,#shape中的维度个数
        "dims": [xx, xx, xx]   # 维度信息
        }
针对动态Batch、动态分辨率(宽高)的场景,输入Tensor的“dims”中Batch数或宽高为-1,表示其动态可变。 例如,输入Tensor的“format”为NCHW,在动态Batch场景下,动态可变的输入Tensor的“dims”为[-1, 3, 224, 224],在动态分辨率场景下,动态可变的输入Tensor的“dims”为[1, 3, -1, -1]。(举例中的斜体部分以实际情况为准。) 
 ret:int,错误码。 
  | 
约束说明  | 
无  |