tf.keras.layers.Permute

Pengsen Ma

于 2021-05-10 21:12:36 发布

阅读量1.8w

点赞数

分类专栏： StyleGAN代码学习

本文链接：https://blog.csdn.net/weixin_43135178/article/details/116610636

版权

10 篇文章

订阅专栏

本文详细介绍了Keras中Permute层的使用方法及其参数设置。Permute层用于改变输入张量的维度顺序，这对于调整数据格式至关重要。文章通过一个具体示例展示了如何使用Permute层来调整输入数据的维度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

更改维度的顺序,维度索引从1开始

Permute：

keras.layers.Permute(dims)

根据给定的模式(dim)置换输入的维度。
例如：

model = Sequential()
modle.add(Permute((2,1),input_shape=(10,64)))
# 现在: model.output_shape == (None, 64,10)
# “None"是batch的维度

参数：

dims: 整数元组，是置换模式，不包含样本维度，索引从1开始，例如(2,1)即置换输入的第1和第2个维度
输入尺寸：
任意。当使用此层作为模型中的第一层时，使用参数input_shape(整数元素，不包括样本数的axis)

输出尺寸：
与输入尺寸相同，但是维度根据指定的模式重新排列。

————————————————
原文链接：https://blog.csdn.net/ChenglinBen/article/details/95864488