set_value

功能说明

将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 训练系列产品

Atlas 推理系列产品

Atlas 200/500 A2推理产品

Atlas A2训练系列产品/Atlas 800I A2推理产品

调用示例

输入Tensor x为：

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

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]]