文档
注册

函数:get_tensor_desc_dim

此接口后续版本会废弃,请使用acl.get_tensor_desc_dim_v2接口。

C函数原型

int64_t aclGetTensorDescDim(const aclTensorDesc *desc, size_t index)

Python函数

num = acl.get_tensor_desc_dim(desc, index)

函数功能

获取tensor描述中指定维度的大小,同步接口。

输入说明

desc:int,aclTensorDesc类型的指针地址。

需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。

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

用户调用acl.get_tensor_desc_dim接口获取Shape维度个数,“index”的取值范围:[0, (shape维度个数-1)]。

返回值说明

num:int,获取Tensor描述中指定维度的大小。

约束说明

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

当前产品无相关内容

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