tf.keras.layers.Permute
| Просмотреть исходный код на GitHub |
Переставляет порядок измерений входных данных в соответствии с заданным шаблоном.
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 (кортеж целых чисел, не включая ось образцов) при использовании этого слоя как первого слоя в модели.
Форма выходных данных:
Такая же, как форма входных данных, но с переставленными измерениями в соответствии с указанным шаблоном.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/Permute