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