跳到主要内容

函数:get_tensor_desc_dim_range

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

父主题: 公共接口

函数:get_tensor_desc_dim_v2

C函数原型aclError aclGetTensorDescDimV2(const aclTensorDesc *desc, size_t index, int64_t *dimSize)
Python函数dim_size, ret = acl.get_tensor_desc_dim_v2(desc, index)
函数功能获取Tensor描述中指定维度的大小。
输入说明desc:int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 index:int,指定获取第几个维度的大小,“index”值从0开始。 用户调用acl.get_tensor_desc_num_dims接口获取Shape维度个数,“index”的取值范围:[0, (*shape维度个数-*1)]。
返回值说明dim_size:int,Tensor描述中“index”指定维度的大小。当返回的“dim_size”参数值为“-1”时,表示维度的大小是动态的。 ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明acl.get_tensor_desc_num_dims接口的返回值为**ACL_UNKNOWN_RANK(0xFFFFFFFFFFFFFFFE)**时,表示动态Shape场景下维度个数未知,则不能调用acl.get_tensor_desc_dim_v2接口获取指定维度的范围。

父主题: 公共接口

函数: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接口获取指定维度的范围。

父主题: 公共接口

函数:set_tensor_desc_name

C函数原型**void aclSetTensorDescName(**aclTensorDesc *desc, const char *name)
Python函数acl.set_tensor_desc_name(desc, name)
函数功能设置TensorDesc的名称,同步接口。
输入说明desc:int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 name:str,Tensor描述的名称。
返回值说明
约束说明

父主题: 公共接口

函数:get_tensor_desc_name

C函数原型**const char *aclGetTensorDescName(**aclTensorDesc *desc)
Python函数name = acl.get_tensor_desc_name(desc)
函数功能获取tensor描述的名称,同步接口。
输入说明desc:int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。
返回值说明name:str,Tensor描述的名称。
约束说明

父主题: 公共接口

函数:set_tensor_storage_shape

备注

须知

此接口后续版本会废弃,请使用acl.set_tensor_shape接口。

C函数原型aclError aclSetTensorStorageShape(aclTensorDesc *desc, int numDims, const int64_t *dims)
Python函数ret = acl.set_tensor_storage_shape(desc, dims)
函数功能调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过acl.set_tensor_storage_shape设置Tensor的实际“dims”信息,当前主要用于PyTorch场景下,同步接口。
输入说明desc:int,aclTensorDesc类型的指针地址。 需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 dims:list,要设置的“dims”维度数。
返回值说明ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: 公共接口

在线提单