Keras 3 API 文档 / 层 API / 合并层 / Concatenate 层

Concatenate 层

[源代码]

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 轴连接的结果。