文档
注册

函数: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”个结果

当前产品无相关内容

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