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是输入 inputTensor 的维度, axis 为负值时与 concat - 图1 等价。默认值为0。
    • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。

返回:联结后的 Tensor ,数据类型和 input 相同。

返回类型:Variable

代码示例

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