通过Tensor数组下标,改变Tensor内容。
__setitem__(index, value)
参数名称 |
输入/输出 |
含义 |
---|---|---|
index |
输入 |
Tensor下标,包括如下类型:
|
value |
输入 |
待设置的值。支持的数据类型: |
Atlas 200/300/500 推理产品
Atlas 训练系列产品
1.目的Tensor的类型只支持scope_ubuf。
2.只会对目的Tensor的第一个元素进行改变。
None
from tbe import tik tik_instance = tik.Tik() data_A = tik_instance.Tensor("float16", (128,), name="data_A", scope=tik.scope_ubuf) scalar_B = tik_instance.Scalar(dtype="float16", name="scalar_B", init_value=2.0) data_A[0] = scalar_B