concat_cn.rst 1.6 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5 6 7
.. _cn_api_fluid_layers_concat:

concat
-------------------------------

.. py:function:: paddle.fluid.layers.concat(input,axis=0,name=None)

8
该OP对输入沿 ``axis`` 轴进行联结。
H
Hao Wang 已提交
9 10

参数:
11 12
    - **input** (list) - 输入是待联结的多维 ``Tensor`` 组成的 ``list`` ,支持的数据类型为:float32、float64、int32、int64。
    - **axis** (int|Variable,可选) - 整数或者形状为[1]的 ``Tensor``,数据类型为 ``int32``。指定对输入Tensor进行运算的轴, ``axis`` 的有效范围是[-R, R),R是输入 ``input`` 中 ``Tensor`` 的维度, ``axis`` 为负值时与 :math:`axis + R` 等价。默认值为0。
Z
zhupengyang 已提交
13
    - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
H
Hao Wang 已提交
14

Z
zhupengyang 已提交
15
返回:联结后的 ``Tensor`` ,数据类型和 ``input`` 相同。
H
Hao Wang 已提交
16

Z
zhupengyang 已提交
17
返回类型:Variable
H
Hao Wang 已提交
18 19 20 21 22

**代码示例**:

.. code-block:: python

Z
zhupengyang 已提交
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
  import paddle.fluid as fluid
  import numpy as np

  in1 = np.array([[1,2,3],
                  [4,5,6]])
  in2 = np.array([[11,12,13],
                  [14,15,16]])
  in3 = np.array([[21,22],
                  [23,24]])
  with fluid.dygraph.guard():
      x1 = fluid.dygraph.to_variable(in1)
      x2 = fluid.dygraph.to_variable(in2)
      x3 = fluid.dygraph.to_variable(in3)
      out1 = fluid.layers.concat(input=[x1,x2,x3], axis=-1)
      out2 = fluid.layers.concat(input=[x1,x2], axis=0)
      print(out1.numpy())
      # [[ 1  2  3 11 12 13 21 22]
      #  [ 4  5  6 14 15 16 23 24]]
      print(out2.numpy())
      # [[ 1  2  3]
      #  [ 4  5  6]
      #  [11 12 13]
      #  [14 15 16]]