函数：get_cur_output_dims

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

√

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

√

Atlas 训练系列产品

√

Atlas 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 200/300/500 推理产品

√

aclError aclmdlGetCurOutputDims(const aclmdlDesc *modelDesc, size_t index, aclIODims *dims)

dims, ret = acl.mdl.get_cur_output_dims(model_desc, index)

根据模型描述信息获取指定的模型输出Tensor的实际维度信息。

model_desc：int，aclmdlDesc类型数据的指针地址。

需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。

index：int，指定获取第几个输出的Dims，index值从0开始。

dims：dict，输出实际维度信息。

• 若Tensor的“name”长度大于127，则在输出dims["name"]时，会将Tensor的“name”转换为“acl_modelId_${id}_input_${index}_${随机字符串} ”格式并在转换后的“name”与原“name”之间建立映射关系，用户可调用acl.mdl.get_tensor_real_name接口，传入转换后的“name”，获取原“name”（若向接口传入原“name”，则获取的还是原“name”）。
  说明：

  如果转换后的Tensor的“name”与模型中已有的Tensor的“name”冲突，则会在转换后的“name”尾部增加“_${随机字符串} ”，否则不会增加随机字符串。

• 若Tensor的“name”长度小于或等于127，则在输出的dims["name"]时，按Tensor的“name”输出。

ret：int，错误码。

• 返回0表示成功。
• 返回其它值表示失败。

当前仅支持通过本接口获取以下场景中的模型输出Tensor的维度信息：

• 通过模型转换设置多档Batch size或分辨率或维度值，实现动态Batch或动态分辨率或动态维度（ND格式）时。
  • 如果用户已调用acl.mdl.set_dynamic_batch_size设置Batch或调用acl.mdl.set_dynamic_hw_size接口设置输入图片的宽高或调用acl.mdl.set_input_dynamic_dims接口设置某动态维度的值，则可通过该接口获取指定模型输出Tensor的实际维度信息。
  • 如果用户未调用acl.mdl.set_dynamic_batch_size接口、acl.mdl.set_dynamic_hw_size接口或acl.mdl.set_input_dynamic_dims接口，则通过该接口可获取最大档的维度信息。
• 固定Shape场景下，通过该接口获取指定的模型输出Tensor的维度信息。