array_write
该OP将输入的变量 x
写入到数组 array
的第i个位置,并返回修改后的LoDTensorArray,如果 array
为None,则创建一个新的LoDTensorArray。常与 array_read OP联合使用对LoDTensorArray进行读写。
参数:
i (Variable) – shape为[1]的1-D Tensor,表示写入到输出数组LoDTensorArray的位置,数据类型为int64
array (Variable,可选) – 指定写入
x
的数组LoDTensorArray。默认值为None, 此时将创建新的LoDTensorArray并作为结果返回
返回类型: Variable
代码示例
Tensor[array_read_0.tmp_0]
shape: [3,2,]
dtype: l
#返回了shape为[3,2]的Tensor,即为上面写入的tmp
#dtype为对应C++数据类型,在不同环境下可能显示值不同,但本质一致
#例如:如果Tensor中数据类型是int64,则对应的C++数据类型为int64_t,所以dtype值为typeid(int64_t).name(),