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)