aclGetTensorDescDimV2

Applicability

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

Description

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

Prototype

aclError aclGetTensorDescDimV2(const aclTensorDesc *desc, size_t index, int64_t *dimSize)

Parameters

Parameter	Input/Output	Description
desc	Input	Pointer to data of the aclTensorDesc type. Call aclCreateTensorDesc to create data of the aclTensorDesc type in advance.
index	Input	Sequence number of the dimension whose size to obtain, indexed starting at 0. Must be in the range of [0, Dimension count – 1]. Call aclGetTensorDescNumDims to obtain the dimension count.
dimSize	Output	Pointer to the size of the dimension specified by index in the tensor description. If -1 is returned, the dimension size is not determinable until runtime.

Parameter

Input/Output

Description

desc

Input

Pointer to data of the aclTensorDesc type.

Call aclCreateTensorDesc to create data of the aclTensorDesc type in advance.

index

Input

Sequence number of the dimension whose size to obtain, indexed starting at 0.

Must be in the range of [0, Dimension count – 1]. Call aclGetTensorDescNumDims to obtain the dimension count.

dimSize

Output

Pointer to the size of the dimension specified by index in the tensor description.

If -1 is returned, the dimension size is not determinable until runtime.

Returns

0 on success; else, failure. For details, see aclError.

Restrictions

If the aclGetTensorDescNumDims call returns ACL_UNKNOWN_RANK, it indicates that the dimension count is unknown in the dynamic shape scenario. In this case, calling aclGetTensorDescDimV2 cannot obtain the size of the specified dimension.

Parent topic: aclTensorDesc