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