On this page
tf.layers.InputSpec
Specifies the ndim, dtype and shape of every input to a layer.
tf.layers.InputSpec(
dtype=None, shape=None, ndim=None, max_ndim=None, min_ndim=None, axes=None
)
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
@classmethod
from_config(
config
)
get_config
get_config()
© 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