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.

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.

Arbitrary.

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)

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.

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.