stack_cn.rst

.. _cn_api_paddle_tensor_arange
stack
-------------------------------

.. py:function:: paddle.tensor.stack(x, axis=0)

该OP沿 axis 轴对输入 x 进行堆叠操作。

**参数**：
        - **x** (Variable|list(Variable)) – 输入 x 可以是单个Tensor，或是多个Tensor组成的列表。如果 x 是一个列表，那么这些Tensor的维度必须相同。 假设输入是N维Tensor [d0,d1,...,dn−1]，则输出变量的维度为N+1维 [d0,d1,...daxis−1,len(x),daxis...,dn−1] 。支持的数据类型: float32，float64，int32，int64。

        - **axis** (int, 可选) – 指定对输入Tensor进行堆叠运算的轴，有效 axis 的范围是: [−(R+1),R+1)]，R是输入中第一个Tensor的rank。如果 axis < 0，则 axis=axis+rank(x[0])+1 。axis默认值为0。

**返回**：堆叠运算后的Tensor，数据类型与输入Tensor相同。输出维度等于 rank(x[0])+1 维。

**返回类型**：Variable

**代码示例**:

.. code-block:: python
   
    import paddle
    import paddle.fluid as fluid
    import numpy as np
    data1 = np.array([[1.0, 2.0,3.0]])
    data2 = np.array([[3.0, 4.0, 5.0]])
    data3 = np.array([[5.0, 6.0,7.0]])
    with fluid.dygraph.guard():
        x1 = fluid.dygraph.to_variable(data1)
        x2 = fluid.dygraph.to_variable(data2)
        x3 = fluid.dygraph.to_variable(data3)
        result = paddle.stack([x1, x2, x3], axis=2)
        # result shape: [3, 1, 2]
        # result value: [[[1.0, 2.0]],
        #                [[3.0, 4.0]],
        #                [[5.0, 6.0]]]