tf.keras.InputSpec

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

Альтернативные имена

Основные альтернативные имена

tf.keras.layers.InputSpec

Альтернативные имена для совместимости при миграции

Подробнее см. в руководстве по миграции.

tf.compat.v1.keras.InputSpec

tf.keras.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 (по одному на каждый входной тензор). Эти объекты позволяют слою проверять совместимость входных данных по структуре, рангу, форме и типу данных для первого аргумента Layer.call.

Запись 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().__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/api_docs/python/tf/keras/InputSpec