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 串聯的結果。