Permute 層

[原始碼]

Permute 類別

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

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

例如，連接 RNN 和卷積網路時很有用。

範例

model = Sequential()
model.add(Permute((2, 1), input_shape=(10, 64)))
# now: model.output_shape == (None, 64, 10)
# note: `None` is the batch dimension

參數

• dims: 整數元組。置換模式不包括樣本維度。索引從 1 開始。例如，(2, 1) 會置換輸入的第一個和第二個維度。

輸入形狀

任意。當將此層用作模型中的第一層時，請使用關鍵字參數 input_shape (整數元組，不包括樣本軸)。

輸出形狀

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