InputSpec
classkeras.InputSpec(
dtype=None,
shape=None,
ndim=None,
max_ndim=None,
min_ndim=None,
axes=None,
allow_last_axis_squeeze=False,
name=None,
optional=False,
)
Specifies the rank, dtype and shape of every input to a layer.
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 for the first argument of Layer.__call__
.
A None
entry in a shape is compatible with any dimension.
Arguments
None
for dynamic axes).
Includes the batch size.True
, allow inputs of rank N+1 as long
as the last axis of the input is 1, as well as inputs of rank N-1
as long as the last axis of the spec is 1.None
values.Example
class MyLayer(Layer):
def __init__(self):
super().__init__()
# The layer will accept inputs with
# shape (*, 28, 28) & (*, 28, 28, 1)
# and raise an appropriate error message otherwise.
self.input_spec = InputSpec(
shape=(None, 28, 28, 1),
allow_last_axis_squeeze=True)