tf.keras.layers.InputSpec

Specifies the rank, dtype and shape of every input to a layer.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.layers.InputSpec, tf.compat.v1.layers.InputSpec

tf.keras.layers.InputSpec(
    dtype=None, shape=None, ndim=None, max_ndim=None, min_ndim=None, axes=None
)

Layers can expose (if appropriate) an input_spec attribute: an instance of InputSpec, or a nested structure of InputSpec instances (one per input tensor). These objects enable the layer to run input compatibility checks for input structure, input rank, input shape, and input dtype.

A None entry in a shape is compatible with any dimension, a None shape is compatible with any shape.

Arguments
dtype	Expected DataType of the input.
shape	Shape tuple, expected shape of the input (may include None for unchecked axes).
ndim	Integer, expected rank of the input.
max_ndim	Integer, maximum rank of the input.
min_ndim	Integer, minimum rank of the input.
axes	Dictionary mapping integer axes to a specific dimension value.

Methods

from_config

View source

@classmethod
from_config(
    config
)

get_config

View source

get_config()