Python tf.keras.layers.Permute用法及代碼示例

根據給定的模式排列輸入的維度。

繼承自：Layer，Module

用法

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

輸入形狀：

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

輸出形狀：

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