Функция Input

Функция Input создает входной слой для функциональной модели Keras. Первым параметром передается форма входных данных (shape) без учета размера батча. Вторым параметром можно передать тип данных (dtype), а также имя слоя (name). Функция возвращает тензор Keras, который используется при построении модели.

Синтаксис

tf.keras.Input(shape=None, [batch_size=None], [dtype=None], [name=None])

Пример

Давайте создадим входной слой с формой (5,) и выведем его параметры:

import tensorflow as tf

inp = tf.keras.Input(shape=(5,))
print(inp)

Результат выполнения кода:

<KerasTensor shape=(None, 5), dtype=float32, sparse=False, name=keras_tensor>

Пример

Давайте создадим входной слой с типом int32 и именем input_layer:

import tensorflow as tf

inp = tf.keras.Input(shape=(5,), dtype=tf.int32, name='input_layer')
print(inp)

Результат выполнения кода:

<KerasTensor shape=(None, 5), dtype=int32, sparse=False, name=input_layer>

Пример

Давайте построим простую функциональную модель с входным слоем и полносвязным слоем:

import tensorflow as tf

inp = tf.keras.Input(shape=(5,))
out = tf.keras.layers.Dense(1)(inp)
model = tf.keras.Model(inputs=inp, outputs=out)
model.summary()

Результат выполнения кода:

Model: "model"
_________________________________________________________________
 Layer (type)				Output Shape			  Param #   
=================================================================
 input_1 (InputLayer)		[(None, 5)]			   0		 
																 
 dense (Dense)			   (None, 1)				 6		 
																 
=================================================================
Total params: 6
Trainable params: 6
Non-trainable params: 0
_________________________________________________________________

Смотрите также

слой InputLayer,
который является слоем входных данных
слой Dense,
который создает полносвязный слой
слой Activation,
который применяет функцию активации
слой Dropout,
который применяет dropout к входным данным