函数:set_tensor_value_range

C函数原型

aclError aclSetTensorValueRange(aclTensorDesc* desc, size_t valueCount, int64_t valueRange[][ACL_TENSOR_VALUE_RANGE_NUM])

Python函数

ret = acl.set_tensor_value_range(desc, value_range)

函数功能

调用acl.create_tensor_desc接口创建tensor描述信息后,可通过本接口设置tensor的数据值的范围。同步接口。

使用场景:部分算子input的值就是该算子的输出Shape,在动态Shape场景下,这个Shape的值就有一个取值范围,在执行算子时,需要在input上设置Shape值的范围(例如:[[16,16],[1,128],[20,20],[1,10]]),pyACL才能正常编译、执行算子。

输入说明

desc : int aclTensorDesc类型的指针。需提前调用函数:create_tensor_desc接口创建aclTensorDesc类型。

value_range: list value_range为每个数据值的范围,用二维数组表示范围。

 ACL_TENSOR_VALUE_RANGE_NUM = 2。

返回值说明

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

注意事项