W3cubDocs

/TensorFlow

tf.keras.layers.Permute

Permutes the dimensions of the input according to a given pattern.

Inherits From: Layer, Operation

Useful e.g. connecting RNNs and convnets.

Args
dims Tuple of integers. Permutation pattern does not include the batch dimension. Indexing starts at 1. For instance, (2, 1) permutes the first and second dimensions of the input.
Input shape
Arbitrary.
Output shape
Same as the input shape, but with the dimensions re-ordered according to the specified pattern.

Example:

x = keras.Input(shape=(10, 64))
y = keras.layers.Permute((2, 1))(x)
y.shape
(None, 64, 10)
Attributes
input Retrieves the input tensor(s) of a symbolic operation.

Only returns the tensor(s) corresponding to the first time the operation was called.

output Retrieves the output tensor(s) of a layer.

Only returns the tensor(s) corresponding to the first time the operation was called.

Methods

from_config

View source

Creates a layer from its config.

This method is the reverse of get_config, capable of instantiating the same layer from the config dictionary. It does not handle layer connectivity (handled by Network), nor weights (handled by set_weights).

Args
config A Python dictionary, typically the output of get_config.
Returns
A layer instance.

symbolic_call

View source

© 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/api_docs/python/tf/keras/layers/Permute