Function: get_tensor_desc_dim_range

C Prototype

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

Python Function

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

Function Usage

Obtains the range of a tensor dimension in the tensor description. Note that [1, -1] indicates that the dimension size has no limit.

Input Description

desc: int, pointer address of aclTensorDesc.

Call acl.create_tensor_desc to create data of the aclTensorDesc type in advance.

index: int, sequence number of the dimension whose size is to be obtained, starting from 0.

Value range: [0, Dimension count – 1]. Call acl.get_tensor_desc_num_dims to obtain the dimension count.

dim_range_num: int, dim_range length. The value must be greater than or equal to 2.

Return Value

dim_range: list, size range of the dimension specified by index in the tensor description.

  • The first element indicates the minimum value of the range, and the second element indicates the maximum value of the range. Only the first two elements in the array are valid.
  • [1, -1] indicates that the dimension size has no limit.

ret: int, error code.

Restrictions

If the return value of acl.get_tensor_desc_num_dims is ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE), the dimension count is unknown. In this case, calling acl.get_tensor_desc_dim_range cannot obtain the range of the specified dimension.