Permute 層

[原始碼]

Permute 類別

keras.layers.Permute(dims, **kwargs)

根據給定的模式置換輸入的維度。

例如,用於連接 RNN 和卷積網路。

參數

  • dims:整數元組。置換模式不包含批次維度。索引從 1 開始。例如,(1, 3, 2) 置換輸入的第二和第三維度。

輸入形狀

任意。

輸出形狀

與輸入形狀相同,但維度根據指定的模式重新排序。

範例

>>> x = keras.Input(shape=(10, 64))
>>> y = keras.layers.Permute((2, 1))(x)
>>> y.shape
(None, 64, 10)