函数: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描述信息的指针地址。 |
约束说明 |
无 |
父主题: 公共接口