文档
注册

函数: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描述信息的指针地址。

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词