函数：set_input_dynamic_dims

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

√

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

√

Atlas 训练系列产品

√

Atlas 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 200/300/500 推理产品

√

modelId

int，模型ID。

可在以下接口成功加载模型后获取到模型ID。

• acl.mdl.load_from_file
• acl.mdl.load_from_mem
• acl.mdl.load_from_file_with_mem
• acl.mdl.load_from_mem_with_mem

dataset

int，表示模型的输入数据。

使用aclmdlDataset类型的数据描述模型推理时的输入数据，输入的内存地址、内存大小用aclDataBuffer类型的数据来描述，aclmdlDataset。

index

int，标识动态维度的输入index。

需调用acl.mdl.get_input_index_by_name接口获取，输入名称固定为“ascend_mbatch_shape_data”。

dims

dict，指定具体某一档的所有维度信息，具体请参见aclmdlIODims。

可以调用acl.mdl.get_input_dynamic_dims接口获取指定模型支持的动态维度档位数以及每一档中的值。

例如：在ATC工具进行模型转换时，使用以下参数：input_shape="data:1,1,40,-1;label:1,-1;mask:-1,-1" ，dynamic_dims="20,20,1,1; 40,40,2,2; 80,60,4,4"

若输入数据的真实维度为（1,1,40,20,1,20,1,1），则dims输入示例如下（name暂不使用）：

dims = {'dimCount': 8, 'name': '', 'dims': [1,1,40,20,1,20,1,1]}

ret

int，错误码，返回0表示成功，返回其它值表示失败。