更新时间:2021-03-18 GMT+08:00
set_as
功能说明
设置或者改变Tensor 的内容。
函数原型
set_as(value, dst_offset=0, src_offset=None)
参数说明
参数名称 |
输入/输出 |
含义 |
---|---|---|
value |
输入 |
待设置的值。支持的数据类型:
|
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])
父主题: Tensor管理
