函数：get_tensor_desc_by_index
C函数原型 |
aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index) |
|---|---|
Python函数 |
tensor_desc = acl.get_tensor_desc_by_index(desc, index) |
函数功能 |
获取算子的指定Index的Tensor描述信息。 |
输入说明 |
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 ：返回指定输入/输出的Tensor描述信息的指针地址。 |
约束说明 |
无 |
