置换层

[源代码]

Permute

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

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

例如,在连接 RNN 和 convnet 时很有用。

示例

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(整数元组,不包括样本轴)。

输出形状

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