tensor_array_to_tensor

    该OP将 input 这个LoDTensorArray中的所有Tensor沿 axis 指定的轴进行拼接(concat)或堆叠(stack)。

    示例:

    • input (Variable) - 输入的LoDTensorArray。支持的数据类型为:float32、float64、int32、int64。

    • axis (int,可选) - 指定对输入Tensor进行运算的轴, axis 的有效范围是[-R, R),R是输入 input 中Tensor的Rank, 为负时与 axis +R 等价。默认值为1。

    返回:Variable的二元组, 包含了两个Tensor。第一个Tensor表示对数组内的元素进行stack或concat的输出结果,数据类型与数组中的Tensor相同;第二个Tensor包含了数组中各Tensor在 axis 维度的大小,数据类型为int32。

    代码示例:

    1. import paddle.fluid as fluid
    2. import numpy as np
    3. x0 = fluid.layers.assign(np.random.rand(2, 2).astype("float32"))
    4. i = fluid.layers.fill_constant(shape=[1], dtype="int64", value=0)
    5. array = fluid.layers.create_array(dtype='float32')
    6. fluid.layers.array_write(x0, i, array)