跳到主要内容

函数:set_tensor_shape_range

函数:set_tensor_format

C函数原型aclError aclSetTensorFormat(aclTensorDesc *desc, aclFormat format)
Python函数ret = acl.set_tensor_format(desc, format)
函数功能调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过acl.set_tensor_format设置昇腾AI处理器内部处理Tensor时的Format信息,当前主要用于PyTorch场景下,同步接口。 (说明:例如,某算子的原始Format为NHWC(原始Shape为4维)或者NDHWC(原始Shape为5维),为了提高数据访问效率,在进行模型转换或者在线构建网络时会自动转换Format,将原始数据Format转化为 昇腾AI处理器 内部计算数据时用的Format(NC1HWC0或者NDC1HWC0),同时进行推导Shape用于内部计算。)
输入说明desc:int,Tensor描述数据的指针地址。 format:int, 指定Tensor描述的格式。
返回值说明ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: 公共接口

函数:set_tensor_const

C函数原型aclError **aclSetTensorConst(**aclTensorDesc *desc, void *dataBuffer, size_t length)
Python函数ret = acl.set_tensor_const**(**desc, dataBuffer, length)
函数功能为算子设置constant输入,指定存放输入Tensor数据的内存地址和地址长度。同步接口。
输入说明desc:int,aclTensorDesc类型的指针地址。需提前调用acl.create_tensor_desc接口创建aclTensorDesc类型。 dataBuffer:int,存放constant输入Tensor数据的内存地址,此处算子输入Tensor数据的内存必须根据应用运行模式来确定。内存申请接口请参见内存管理
- 应用运行在Host时,此处需申请Host上的内存。
- 应用运行在Device时,此处需申请Device上的内存。
length:int,内存地址的长度,单位是Byte。
返回值说明ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: 公共接口

函数:set_tensor_shape_range

C函数原型aclError aclSetTensorShapeRange(aclTensorDesc* desc, size_t dimsCount, int64_t dimsRange[][ACL_TENSOR_SHAPE_RANGE_NUM]****)
Python函数ret = acl.set_tensor_shape_range(desc, dims_range)
函数功能调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过本接口设置Tensor的各个维度的取值范围。同步接口。 使用场景:动态Shape的算子,其输入Shape中变化维度用-1表示,但每个变化维度的范围是不一样的,需要显式设置。 如Shape为[16, -1, 20, -1],对应的shape range可以是[[16, 16], [1, 128], [20, 20], [1, 10]],表示第一维的Shape范围固定为16,第二维的Shape范围为1到128,第三维的Shape范围固定为20,第四维的Shape范围为1到10。
输入说明desc:int,Tensor描述数据的指针地址。 dims_range:list,“dims_range”为每个维度的取值范围,用二维list表示范围。
返回值说明ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: 公共接口

函数:set_tensor_dynamic_input

C函数原型aclError aclSetTensorDynamicInput(aclTensorDesc *desc, const char *dynamicInputName)
Python函数ret = acl.set_tensor_dynamic_input(desc, dynamic_input_name)
函数功能调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过acl.set_tensor_dynamic_input接口设置算子动态输入的动态输入属性的名称。
输入说明desc:int,Tensor描述数据的指针地址。 dynamic_input_name:int,设置“desc”中的动态输入属性的名称。
返回值说明ret:int,错误码。
- 返回0表示成功。
- 返回其它值表示失败。
约束说明

父主题: 公共接口

函数:get_tensor_desc_by_index

C函数原型aclTensorDesc *aclGetTensorDescByIndex(aclTensorDesc *desc, size_t index)
Python函数tensor_desc = acl.get_tensor_desc_by_index(desc, index)
函数功能获取算子的指定Index的Tensor描述信息。
输入说明desc:int,Tensor描述数据的指针地址。调用acl.mdl.create_and_get_op_desc接口。获取算子所有输入/输出的Tensor描述作为本接口的输入。 index:int,指定获取第几个维度的大小,“index”值从0开始。 用户调用acl.mdl.create_and_get_op_desc接口获取的算子输入/输出的数量后,该“index”的取值范围:[0, (算子输入/输出的数量 - 1)]。
返回值说明tensor_desc :返回指定输入/输出的Tensor描述信息的指针地址。
约束说明

父主题: 公共接口

函数:get_tensor_desc_address

C函数原型void *aclGetTensorDescAddress(const aclTensorDesc *desc)
Python函数addr = acl.get_tensor_desc_address(desc)
函数功能获取指定算子输入/输出的Tensor数据的指针地址。
输入说明desc:int,Tensor描述信息的指针地址。调用acl.get_tensor_desc_by_index接口获取算子所有输入/输出的Tensor描述,作为本接口的输入。
返回值说明addr:int,返回指定算子输入/输出的Tensor数据的指针地址。
约束说明

父主题: 公共接口

在线提单