函数: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描述中指定维度的大小。 |
约束说明 |
无 |
父主题: aclmdlTensorDesc