tf.layers.InputSpec

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

Every layer should expose (if appropriate) an input_spec attribute: a list of instances of InputSpec (one per input tensor).

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

get_config

View source

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/layers/InputSpec