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