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()

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/InputSpec