提交 1a4575f4 编写于 作者: Z zhupengyang 提交者: cyj1986

fix concat doc (#1274)

* fix concat doc
test=document_preview

* fix
test=document_preview

* fix concat attr name
test=document_preview
上级 cee696ba
...@@ -5,35 +5,41 @@ concat ...@@ -5,35 +5,41 @@ concat
.. py:function:: paddle.fluid.layers.concat(input,axis=0,name=None) .. py:function:: paddle.fluid.layers.concat(input,axis=0,name=None)
**Concat** 该OP对输入沿 ``axis`` 轴进行联结
这个函数将输入连接在前面提到的轴上,并将其作为输出返回。
参数: 参数:
- **input** (list)-将要联结的张量列表 - **input** (list) - 输入待联结的多维 ``Tensor`` 组成的 ``list`` ,支持的数据类型为:float32、float64、int32、int64。
- **axis** (int)-数据类型为整型的轴,其上的张量将被联结 - **axis** (int,可选) - 指定对输入Tensor进行运算的轴, ``axis`` 的有效范围是[-R, R),R是输入 ``input`` 中 ``Tensor`` 的Rank, ``axis`` 为负时与 ``axis`` +R 等价。默认值为0。
- **name** (str|None)-该层名称(可选)。如果设为空,则自动为该层命名 - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None
返回:输出的联结变量 返回:联结后的 ``Tensor`` ,数据类型和 ``input`` 相同。
返回类型:变量(Variable) 返回类型:Variable
**代码示例**: **代码示例**:
.. code-block:: python .. code-block:: python
import paddle.fluid as fluid
a = fluid.layers.data(name='a', shape=[2, 13], dtype='float32')
b = fluid.layers.data(name='b', shape=[2, 3], dtype='float32')
c = fluid.layers.data(name='c', shape=[2, 2], dtype='float32')
d = fluid.layers.data(name='d', shape=[2, 5], dtype='float32')
out = fluid.layers.concat(input=[a, b, c, d], axis=2)
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]]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册