W3cubDocs

/TensorFlow Python

tf.layers.InputSpec

Class InputSpec

Aliases:

  • Class tf.keras.layers.InputSpec
  • Class tf.layers.InputSpec

Defined in tensorflow/python/layers/base.py.

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

__init__

__init__(
    dtype=None,
    shape=None,
    ndim=None,
    max_ndim=None,
    min_ndim=None,
    axes=None
)

Initialize self. See help(type(self)) for accurate signature.

© 2018 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/api_docs/python/tf/layers/InputSpec