Keras 3 API文档 / Layers API / 重塑层 / Permute layer

置换层

[源代码]

Permute

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

根据给定的模式置换输入的维度。

例如,可用于连接 RNN 和 CNN。

参数

  • dims: 整型元组。置换模式不包含批次维度。索引从 1 开始。例如,(1, 3, 2) 将输入的第二个和第三个维度置换。

输入形状

任意。

输出形状

与输入形状相同,但维度会根据指定模式重新排序。

示例

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