函数：get_tensor_desc_dim

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

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

此接口后续版本会废弃，请使用acl.get_tensor_desc_dim_v2接口。

C函数原型

int64_t aclGetTensorDescDim(const aclTensorDesc *desc, size_t index)

python函数

dim_size = acl.get_tensor_desc_dim(desc, index)

参数名	说明
desc	int，aclTensorDesc类型的指针地址。需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。
index	int，指定获取第几个维度的大小，“index”值从0开始。用户调用acl.get_tensor_desc_dim接口获取Shape维度个数，“index”的取值范围：[0, (shape维度个数-1)]。

参数名

说明

desc

int，aclTensorDesc类型的指针地址。

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

index

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

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

返回值	说明
dim_size	int，获取Tensor描述中指定维度的大小。

父主题： aclmdlTensorDesc