Function: get_tensor_desc_by_index
C Prototype |
aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index) |
|---|---|
Python Function |
tensor_desc = acl.get_tensor_desc_by_index(desc, index) |
Function Usage |
Obtains a tensor description based on the specified index of an operator. |
Input Description |
desc: int, pointer address of the tensor description data. Call acl.mdl.create_and_get_op_desc to obtain the tensor description of all inputs or all outputs of the operator. index: int, sequence number of the dimension whose size is to be obtained, starting from 0. Value range: [0, Operator input or output count – 1]. Call acl.mdl.create_and_get_op_desc to obtain the operator input or output count. |
Return Value |
tensor_desc: pointer address of the tensor description of a specified input or output. |
Restrictions |
None |