aclGetTensorDescDimV2

Applicable Products

Product	Supported
Ascend 950PR/Ascend 950DT	√
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	√

Function Usage

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	Index of the dimension size to obtain, indexed from 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

Index of the dimension size to obtain, indexed from 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 return value of aclGetTensorDescNumDims is ACL_UNKNOWN_RANK, the dimension count is unknown. In this case, calling aclGetTensorDescDimV2 cannot obtain the size of the specified dimension.

Parent topic: aclTensorDesc