函数：get_tensor_desc_dim_v2

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

√

Atlas 训练系列产品

√

Atlas 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 200/300/500 推理产品

√

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

dim_size, ret = acl.get_tensor_desc_dim_v2(desc, index)

获取Tensor描述中指定维度的大小。

desc：int，aclTensorDesc类型的指针地址。

需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。

index：int，指定获取第几个维度的大小，“index”值从0开始。

用户调用acl.get_tensor_desc_num_dims接口获取Shape维度个数，“index”的取值范围：[0, (shape维度个数-1)]。

dim_size：int，Tensor描述中“index”指定维度的大小。当返回的“dim_size”参数值为“-1”时，表示维度的大小是动态的。

ret：int，错误码。

• 返回0表示成功。
• 返回其它值表示失败。

当acl.get_tensor_desc_num_dims接口的返回值为ACL_UNKNOWN_RANK（0xFFFFFFFFFFFFFFFE）时，表示动态Shape场景下维度个数未知，则不能调用acl.get_tensor_desc_dim_v2接口获取指定维度的范围。