set_as

功能说明

设置或者改变Tensor 的内容。

函数原型

set_as(value, dst_offset=0, src_offset=None)

参数说明

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas 推理系列产品AI Core

Atlas 推理系列产品Vector Core

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

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

Atlas 200I/500 A2 推理产品

注意事项

1. Scalar赋值给Tensor时必须保证两者数据类型一致。
2. 目的Tensor的类型只支持scope_ubuf。
3. 只会对目的Tensor的第一个元素进行改变。
4. 支持Expr赋值给Tensor(int/uint)，要求Expr中的Scalar类型为Scalar(int/uint)，立即数类型为(int/float)。
5. Atlas 200/300/500 推理产品，不支持任何类型的Expr赋值给Scalar(float16/float32)。

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

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

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

   Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件，不支持任何类型的Expr赋值给Scalar(float16)。

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

返回值

无

调用示例

from tbe import tik
tik_instance = tik.Tik()
data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_ubuf)
data_B = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_ubuf)
data_A[0].set_as(data_B[0:1])