tf.keras.layers.Permute

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

Inherits From: Layer, Operation

tf.keras.layers.Permute(
    dims, **kwargs
)

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, (1, 3, 2) permutes the second and third 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

@classmethod
from_config(
    config
)

Creates an operation from its config.

This method is the reverse of get_config, capable of instantiating the same operation from the config dictionary.

Note: If you override this method, you might receive a serialized dtype config, which is a dict. You can deserialize it as follows:

if "dtype" in config and isinstance(config["dtype"], dict):
    policy = dtype_policies.deserialize(config["dtype"])

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

Returns
An operation instance.