Function: get_output_dims
C Prototype |
aclError aclmdlGetOutputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims) |
|---|---|
Python Function |
dims, ret = acl.mdl.get_output_dims(model_desc, index) |
Function Usage |
Obtains tensor dimension information of an output based on the model description.
|
Input 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 input to be obtained, starting from 0. |
Return Value |
dims: dict, maximum supported combinations for dynamic-batch or dynamic-size scenarios based on the dictionary array. For details, see aclmdlIODims. dims = {
"name": xxx, #tensor name
"dimCount": xxx, #Number of dimensions in shape
"dims": [xx, xx, xx] # dimension information
}
ret: int, error code.
|
Restrictions |
None |