concat
paddle.fluid.layers.
concat
(input, axis=0, name=None)[源代码]
该OP对输入沿 axis
轴进行联结。
- 参数:
- input (list) - 输入是待联结的多维
Tensor
组成的list
,支持的数据类型为:float32、float64、int32、int64。 - axis (int|Variable,可选) - 整数或者形状为[1]的
Tensor
,数据类型为int32
。指定对输入Tensor进行运算的轴,axis
的有效范围是[-R, R),R是输入input
中Tensor
的维度,axis
为负值时与 等价。默认值为0。 - name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。
- input (list) - 输入是待联结的多维
返回:联结后的 Tensor
,数据类型和 input
相同。
返回类型:Variable
代码示例:
- 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]]