下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

函数:get_tensor_desc_dim_range

C函数原型

aclError aclGetTensorDescDimRange(const aclTensorDesc *desc, size_t index, size_t dimRangeNum, int64_t *dimRange)

Python函数

dim_range, ret = acl.get_tensor_desc_dim_range(desc, index, dim_range_num)

函数功能

获取tensor描述中指定维度的范围,[1,-1]表示全shape范围。

输入说明

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

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

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

用户调用acl.get_tensor_desc_num_dims接口获取shape维度个数,这个Index的取值范围:[0, (shape维度个数-1)]。

dim_range_num:int,dim_range的长度,该值必须大于或等于2。

返回值说明

dim_range:list,tensor描述中index指定维度的Shape范围。

  • 第一个元素值表示Shape范围的最小值,第二个元素值表示Shape范围的最大值。该数组中仅前2个元素值有效。
  • 当dim_range数组的值为[1,-1]时,表示全Shape范围。

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

acl.get_tensor_desc_num_dims接口的返回值为ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE)时,表示动态Shape场景下维度个数未知,则不能调用acl.get_tensor_desc_dim_range接口获取指定维度的范围。

注意事项

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

当前产品无相关内容

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