tf.TensorArray

 TensorArray可以看做是具有动态size功能的Tensor数组。通常都是跟while_loop或map_fn结合使用。

ta = tf.TensorArray(size, dtype, dynamic_size=True)  # dynamic_size指定数组长度可变
ta.stack(name=None)  # 将TensorArray中元素叠起来当做一个Tensor输出
ta.unstack(value, name=None)  # 可以看做是stack的反操作，输入Tensor，输出一个新的TensorArray对象，value按照其axis=0进行分割
ta.write(index, value, name=None)  # 指定index位置写入Tensor
ta.read(index, name=None)  # 读取指定index位置的Tensor