tf.keras.layers.InputSpec
Указывает ранг, тип данных и форму каждого входного значения для слоя.
tf.keras.layers.InputSpec(
dtype=None, shape=None, ndim=None, max_ndim=None, min_ndim=None, axes=None
)
Слои могут (при необходимости) экспонировать атрибут input_spec: экземпляр InputSpec, или вложенную структуру экземпляров InputSpec (по одному на тензор входных данных). Эти объекты позволяют слою выполнять проверки совместимости входных данных по структуре, рангу, форме и типу входных данных.
Элемент None в форме совместим с любой размерностью, форма None совместима с любой формой.
| Аргументы | |
|---|---|
dtype | Ожидаемый тип данных входных данных. |
shape | Кортеж формы, ожидаемая форма входных данных (может включать None для необязательных осей). |
ndim | Целое число, ожидаемый ранг входных данных. |
max_ndim | Целое число, максимальный ранг входных данных. |
min_ndim | Целое число, минимальный ранг входных данных. |
axes | Словарь, сопоставляющий целочисленные оси со значением конкретной размерности. |
Методы
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.3/api_docs/python/tf/keras/layers/InputSpec