tf.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
