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