函数:set_tensor_value_range
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
调用acl.create_tensor_desc接口创建Tensor描述信息后,可通过本接口设置Tensor的数据值的范围。
使用场景:部分算子input的值就是该算子的输出Shape,在动态Shape场景下,这个Shape的值就有一个取值范围,在执行算子时,需要在input上设置Shape值的范围(例如:[[16, 16], [1, 128], [20, 20], [1, 10]]),才能正常执行算子。
函数原型
- C函数原型
1
aclError aclSetTensorValueRange(aclTensorDesc* desc, size_t valueCount, int64_t valueRange[][ACL_TENSOR_VALUE_RANGE_NUM])
- python函数
1
ret = acl.set_tensor_value_range(desc, value_range)
参数说明
参数名 |
说明 |
---|---|
desc |
int,aclTensorDesc类型的指针地址。需提前调用函数:create_tensor_desc接口创建aclTensorDesc类型。 |
value_range |
list,用二维列表表示每个数据值的范围。 ACL_TENSOR_VALUE_RANGE_NUM = 2 |
返回值说明
返回值 |
说明 |
---|---|
ret |
int,错误码。
|
父主题: aclmdlTensorDesc