设置或者改变Scalar的值。
set_as(value, src_offset=None)
参数名称 |
输入/输出 |
含义 |
---|---|---|
value |
输入 |
待设置的值,可以是: |
src_offset |
输入 |
保留参数,不建议使用。 |
昇腾310 AI处理器
昇腾910 AI处理器
昇腾310P AI处理器AI Core
昇腾310P AI处理器Vector Core
昇腾910 AI处理器,不支持任何类型的Expr赋值给Scalar(float16)。
昇腾310P AI处理器AI Core,不支持任何类型的Expr赋值给Scalar(float16)。
昇腾310P AI处理器Vector Core,不支持任何类型的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)