函数：get_tensor_desc_dim_range

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

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

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)

参数名	说明
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。

参数名

说明

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表示成功。返回其它值表示失败。

返回值

说明

dim_range

list，Tensor描述中“index”指定维度的Shape范围。

ret

int，错误码。

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

父主题： aclmdlTensorDesc