tf.keras.layers.InputSpec

Указывает ранг, тип данных и форму каждого входного значения слоя.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

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,
    allow_last_axis_squeeze=False, name=None
)

Слои могут (при необходимости) экспонировать атрибут input_spec: экземпляр InputSpec, или вложенную структуру экземпляров InputSpec (по одному на каждый входной тензор). Эти объекты позволяют слою выполнять проверки совместимости входных данных по структуре, рангу, форме и типу данных.

Значение None в форме совместимо с любой размерностью, а форма None совместима с любой формой.

Аргументы
`dtype`	Ожидаемый тип данных входных данных.
`shape`	Кортеж формы, ожидаемая форма входных данных (может содержать None для неопределенных осей). Включает размер пакета.
`ndim`	Целое число, ожидаемый ранг входных данных.
`max_ndim`	Целое число, максимальный ранг входных данных.
`min_ndim`	Целое число, минимальный ранг входных данных.
`axes`	Словарь, сопоставляющий целочисленные оси со значением конкретной размерности.
`allow_last_axis_squeeze`	Если True, то допускаются входные данные ранга N+1, при условии, что последняя ось входных данных равна 1, а также входные данные ранга N-1, при условии, что последняя ось спецификации равна 1.
`name`	Ожидаемый ключ, соответствующий этому входу при передаче данных как словаря.

Пример:

class MyLayer(Layer):
    def __init__(self):
        super(MyLayer, self).__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)

Методы

`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/r2.4/api_docs/python/tf/keras/layers/InputSpec