set_as

功能说明

设置或者改变Scalar的值。

函数原型

set_as(value, src_offset=None)

参数说明

表1 参数说明
参数名称	输入/输出	含义
value	输入	待设置的值，可以是：立即数（int, float） Scalar Expr：包括Scalar变量、立即数组成的Expr Tensor中的某个值, Tesnor 的scope有如下约束： Atlas 训练系列产品，Tensor的scope只能是scope_ubuf Atlas 推理系列产品AI Core，Tensor的scope可以是scope_ubuf和scope_gm Atlas 推理系列产品Vector Core，Tensor的scope可以是scope_ubuf和scope_gm Atlas A2 训练系列产品/Atlas A2 推理系列产品 Tensor的scope可以是scope_ubuf和scope_gm Atlas A3 训练系列产品/Atlas A3 推理系列产品 Tensor的scope可以是scope_ubuf和scope_gm Atlas 200I/500 A2 推理产品，Tensor的scope可以是scope_ubuf和scope_gm
src_offset	输入	保留参数，不建议使用。

支持的型号

Atlas 训练系列产品

Atlas 推理系列产品AI Core

Atlas 推理系列产品Vector Core

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

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas 200I/500 A2 推理产品

注意事项

Atlas 训练系列产品，不支持任何类型的Expr赋值给Scalar(float16)。

Atlas 推理系列产品AI Core，不支持任何类型的Expr赋值给Scalar(float16)。

Atlas 推理系列产品Vector Core，不支持任何类型的Expr赋值给Scalar(float16)。

Atlas A2 训练系列产品/Atlas A2 推理系列产品，不支持任何类型的Expr赋值给Scalar(float16)。

Atlas A3 训练系列产品/Atlas A3 推理系列产品，不支持任何类型的Expr赋值给Scalar(float16)。

Atlas 200I/500 A2 推理产品，不支持任何类型的Expr赋值给Scalar(float16)。

返回值

无

调用示例

from tbe import tik
tik_instance = tik.Tik()
#立即数：整数
index_reg = tik_instance.Scalar(dtype = "int32")
index_reg.set_as(10)   

#立即数：float
index_reg2 = tik_instance.Scalar(dtype = "float16")
index_reg2.set_as(10.2) 

#Scalar变量
index_reg3 = tik_instance.Scalar(dtype = "float16")
index_reg3.set_as(index_reg2)  

#Tensor某个值
data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_ubuf)
index_reg3.set_as(data_A[0])//tensor的某个值

#Expr
index_reg4 = tik_instance.Scalar(dtype = "int32")
index_reg4.set_as(index_reg+20)

父主题： Scalar管理