tf.keras.layers.Permute
| Просмотреть исходный код на GitHub |
Переупорядочивает размерности входных данных в соответствии с заданным шаблоном.
Унаследовано от: Layer, Module
tf.keras.layers.Permute(
dims, **kwargs
)
Полезно, например, при соединении 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
| Аргументы | |
|---|---|
dims | Кортеж целых чисел. Шаблон перестановки не включает размерность образцов. Нумерация начинается с 1. Например, (2, 1) переупорядочивает первую и вторую размерности входных данных. |
Форма входных данных:
Произвольная. Используйте ключевой аргумент input_shape (кортеж целых чисел, не включающий ось образцов) при использовании этого слоя в качестве первого слоя в модели.
Форма выходных данных:
Такая же, как форма входных данных, но с переупорядоченными размерностями в соответствии с заданным шаблоном.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/layers/Permute