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

拼接层

[source]

Concatenate class

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