Function: get_tensor_desc_dim_v2

Applicability

Product	Supported (√/x)
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	√
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

Function Usage

Obtains the size of a specified dimension of a tensor description.

Prototype

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)

Parameter Description

Parameter	Description
desc	Int, pointer address of the aclTensorDesc type. 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.

Parameter

Description

desc

Int, pointer address of the aclTensorDesc type.

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 Description

Return Value	Description
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. 0: success. Other values: failure.

Return Value

Description

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.

0: success.
Other values: failure.

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.

Parent topic: aclmdlTensorDesc