函数:get_tensor_desc_dim_range
产品支持情况
|
产品 |
是否支持 |
|---|---|
|
Atlas 350 加速卡 |
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
|
|
√ |
功能说明
获取Tensor描述中指定维度的范围,[1, -1]表示全Shape范围。
函数原型
- C函数原型
1aclError aclGetTensorDescDimRange(const aclTensorDesc *desc, size_t index, size_t dimRangeNum, int64_t *dimRange)
- python函数
1dim_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。 |
返回值说明
约束说明
当acl.get_tensor_desc_num_dims接口的返回值为ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE)时,表示动态Shape场景下维度个数未知,则不能调用acl.get_tensor_desc_dim_range接口获取指定维度的范围。
父主题: aclmdlTensorDesc