当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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


根据给定的模式排列输入的维度。

继承自:LayerModule

用法

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

参数

  • dims 整数元组。排列模式不包括样本维度。索引从 1 开始。例如,(2, 1) 置换输入的第一维和第二维。

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

输出形状:

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

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.Permute。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。