功能说明

将Tensor中满足特定条件的元素，设置为指定的值，详细功能介绍可参见调用示例。

函数原型

set_value(tensor, condition, value)

参数说明

tensor：输入Tensor，tvm.Tensor类型。
支持的数据类型：float16, int16, uint16, float32, int32, uint32。
condition：输入的lambda表达式，用于描述set_value的作用范围，即描述输入Tensor中满足什么条件的元素值需要进行修改。
value：指定的数值。支持const、variable或lambda表达式。

返回值

wrapped_tensor：执行set_value之后的Tensor，tvm.Tensor类型

约束说明

无。

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

调用示例

输入Tensor x为：

x = [[1,2,3],
     [4,5,6],
     [7,8,9]]

调用set_value接口将第1维索引大于1，且第二维索引大于0的元素值设置为“0”，代码示例如下所示：

from tbe import tvm
from tbe import dsl
x = tvm.placeholder(（3，3）, dtype=dtype, name="x")
set_valued_tensor = dsl.set_value(x, lambda *i: tvm.all(i[0] > 1, i[1] > 0), 0)

以上代码的功能为指定x[2][1]和x[2][2]的值为“0”，输出结果如下所示：

set_valued_tensor = [[1,2,3],
                     [4,5,6],
                     [7,0,0]]