TensorFlow tf.keras.layers.Permute

最新推荐文章于 2023-04-15 13:00:30 发布

Claroja

最新推荐文章于 2023-04-15 13:00:30 发布

阅读量3.2k

点赞数 2

CC 4.0 BY-SA版权

文章标签： tensorflow

本文链接：https://blog.youkuaiyun.com/claroja/article/details/99581421

本文介绍如何使用Keras中的Permute层调整张量的维度顺序。通过一个具体示例，展示如何交换输入张量的第一维度和第二维度，从而改变模型的输出形状。

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

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

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|(int,int)


参考：
[官网](https://tensorflow.google.cn/api_docs/python/tf/keras/layers/Permute?hl=en#class_permute)