Function: get_tensor_desc_dim_v2
C Prototype |
aclError aclGetTensorDescDimV2(const aclTensorDesc *desc, size_t index, int64_t *dimSize) |
|---|---|
Python Function |
dim_size, ret = acl.get_tensor_desc_dim_v2(desc, index) |
Function Usage |
Obtains the size of a specified dimension of a tensor description. |
Input Description |
desc: int, pointer address of aclTensorDesc. Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance. index: int, sequence number of the dimension whose size is to be obtained, starting from 0. Value range: [0, Dimension count – 1]. Call acl.get_tensor_desc_num_dims to obtain the dimension count. |
Return Value |
dim_size: int, size of the dimension specified by index in the tensor description. If -1 is returned, the dimension size is not determinable until runtime. ret: int, error code.
|
Restrictions |
If the return value of acl.get_tensor_desc_num_dims is ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE), the dimension count is unknown. In this case, calling acl.get_tensor_desc_dim_v2 cannot obtain the range of the specified dimension. |