Keras 3 API 文档 / 层 API / 合并层 / 拼接层

拼接层

[源代码]

Concatenate

keras.layers.Concatenate(axis=-1, **kwargs)

连接一系列输入。

它接收一系列张量作为输入,所有张量除了拼接轴外形状都相同,并返回一个单个张量,它是所有输入的拼接结果。

示例

>>> x = np.arange(20).reshape(2, 2, 5)
>>> y = np.arange(20, 30).reshape(2, 1, 5)
>>> keras.layers.Concatenate(axis=1)([x, y])

在 Keras 模型中的用法

>>> x1 = keras.layers.Dense(8)(np.arange(10).reshape(5, 2))
>>> x2 = keras.layers.Dense(8)(np.arange(10, 20).reshape(5, 2))
>>> y = keras.layers.Concatenate()([x1, x2])

参数

  • axis: 拼接的轴。
  • **kwargs: 标准层关键字参数。

返回值

一个张量,它是沿着 axis 轴拼接输入的结果。