Function: get_cur_output_dims

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

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

Obtains the actual shape of an output tensor based on the model description.

model_desc: int, pointer address of the data of the aclmdlDesc type.

Call acl.mdl.create_desc to create data of the aclmdlDesc type in advance.

index: int, sequence number of the output whose Dims is to be obtained, indexing starting at 0.

dims: dict, actual shape.

• If the length of the tensor name is greater than 127, when dims["name"] is output, pyACL converts the tensor name to the acl_modelId_${id}_input_${index}_${random character string} format and establishes a mapping between the converted name and the original name, you can call acl.mdl.get_tensor_real_name to transfer the converted name and obtain the original name. (If the original name is passed to the API, the original name is obtained.)
  NOTE:

  If the name of the converted tensor conflicts with the name of an existing tensor in the model, _${random string} is added to the end of the converted name. Otherwise, no random string is added.

• If the length of the tensor name is no longer than 127 characters, the tensor name is output when dims["name"] is output.

ret: int, error code.

• 0 indicates success.
• Other values indicate failure.

Currently, this API can be used to obtain the shape of the model output tensors only in the following scenarios:

• Set the batch size, resolution, or dimension value through model conversion to implement dynamic batch, dynamic image size, or dynamic dimension (ND format).
  • If you have set batch by calling acl.mdl.set_dynamic_batch_size, set the width and height of the input image by calling acl.mdl.set_dynamic_hw_size, or set the value of a dynamic dimension by calling acl.mdl.set_input_dynamic_dims, you can call this API to obtain the actual dimension information of the output tensor of a specified model.
  • If the acl.mdl.set_dynamic_batch_size, acl.mdl.set_dynamic_hw_size, or acl.mdl.set_input_dynamic_dims API is not called, this interface can be used to obtain the dimension information of the maximum dimensions.
• Static-shape: This API is used to obtain the shape of the output tensors of a model.