Keras 3 API 文档 / 层 API / 重塑层 / 转换层

转换层

[源代码]

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)