Атрибут input
Атрибут input класса Model возвращает входной тензор модели.
Если модель имеет один вход, атрибут вернет один тензор KerasTensor.
Если модель имеет несколько входов, атрибут вернет список тензоров.
Атрибут доступен только после того, как модель была построена,
то есть после вызова метода build или после первого вызова модели на данных.
Атрибут не принимает параметров. Он возвращает объект KerasTensor
или список объектов KerasTensor в зависимости от архитектуры модели.
Синтаксис
model.input
Пример
Давайте создадим простую модель с одним входом
и посмотрим на содержимое атрибута input:
import tensorflow as tf
model = tf.keras.Sequential([
tf.keras.layers.Dense(3, input_shape=(5,))
])
model.build()
print(model.input)
Результат выполнения кода:
<KerasTensor shape=(None, 5), dtype=float32, sparse=False, name=keras_tensor>
Пример
Давайте создадим модель с несколькими входами
и проверим, что атрибут input вернет список тензоров:
import tensorflow as tf
input_a = tf.keras.Input(shape=(5,))
input_b = tf.keras.Input(shape=(3,))
concat = tf.keras.layers.Concatenate()([input_a, input_b])
output = tf.keras.layers.Dense(1)(concat)
model = tf.keras.Model(inputs=[input_a, input_b], outputs=output)
print(model.input)
Результат выполнения кода:
[<KerasTensor shape=(None, 5), dtype=float32, sparse=False, name=keras_tensor>, <KerasTensor shape=(None, 3), dtype=float32, sparse=False, name=keras_tensor_1>]
Пример
Давайте выведем форму входного тензора модели:
import tensorflow as tf
model = tf.keras.Sequential([
tf.keras.layers.Dense(3, input_shape=(5,))
])
model.build()
print(model.input.shape)
Результат выполнения кода:
(None, 5)
