函数:get_input_dynamic_dims
C函数原型 |
aclError aclmdlGetInputDynamicDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims, size_t gearCount) |
---|---|
Python函数 |
dims_out, ret = acl.mdl.get_input_dynamic_dims(model_desc, index, gear_count) |
函数功能 |
根据模型描述信息获取模型的输入所支持的动态维度信息。 |
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,预留参数,当前未使用,固定设置为-1。 gear_count:int,模型支持的动态维度档位数,需要先通过acl.mdl.get_input_dynamic_gear_count接口获取。 |
返回值说明 |
dims_out:list,获取输入的动态维度信息。 “dims_out”参数是一个列表,模型有几个输入,则该列表就有几个元素。列表中的每个元素都是一个aclmdlIODims的字典,“aclmdlIODims”字典中的dims是一个列表,该列表中的每个元素对应每一档中的具体值。 例如: dims_out, ret = acl.mdl.get_input_dynamic_dims(model_desc, -1, gear_count)
ret:int,错误码。
|
约束说明 |
只有在模型转换时通过“--dynamic_dims”参数设置了分档信息后,才可以调用该接口获取动态维度信息。 例如,模型有三个输入,分别为data(1, 1, 40, -1),label(1, -1),mask(-1, -1) , 其中-1表示动态可变。在模型转换时,“--dynamic_dims”参数的配置示例为:--dynamic_dims="20,20,1,1; 40,40,2,2; 80,60,4,4",则通过本接口获取的动态维度信息为(aclmdlIODims字典内的“name”暂不使用): |