更新时间:2021-03-18 GMT+08:00
分享

set_as

功能说明

设置或者改变Tensor 的内容。

函数原型

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

参数说明

表1 参数说明

参数名称

输入/输出

含义

value

输入

待设置的值。支持的数据类型:

  • Tensor 类型,要求Tensor 只有一个元素。
  • Scalar变量。
  • Expr:Scalar和立即数组成的Expr,但是Scalar和立即数不能是float 类型。

dst_offset

输入

保留参数,不建议使用。

src_offset

输入

保留参数,不建议使用。

注意事项

返回值

调用示例

from te 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])
分享:

    相关文档

    相关产品

close