函数：get_tensor_desc_by_index

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

获取算子的指定Index的Tensor描述信息。

C函数原型

aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index)

python函数

tensor_desc = acl.get_tensor_desc_by_index(desc, index)

参数名	说明
desc	int，Tensor描述数据的指针地址。调用acl.mdl.create_and_get_op_desc接口，获取算子所有输入/输出的Tensor描述作为本接口的输入。
index	int，指定获取第几个维度的大小，“index”值从0开始。调用acl.mdl.create_and_get_op_desc接口获取的算子输入/输出的数量后，该“index”的取值范围：[0, (算子输入/输出的数量 - 1)]。

参数名

说明

desc

int，Tensor描述数据的指针地址。

调用acl.mdl.create_and_get_op_desc接口，获取算子所有输入/输出的Tensor描述作为本接口的输入。

index

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

调用acl.mdl.create_and_get_op_desc接口获取的算子输入/输出的数量后，该“index”的取值范围：[0, (算子输入/输出的数量 - 1)]。

返回值	说明
tensor_desc	int，返回指定输入/输出的Tensor描述信息的指针地址。

父主题： aclmdlTensorDesc