tf.keras.layers.InputLayer
| Просмотреть исходный код на GitHub |
Слой, используемый в качестве точки входа в сеть (граф слоёв).
tf.keras.layers.InputLayer(
input_shape=None, batch_size=None, dtype=None, input_tensor=None, sparse=False,
name=None, ragged=False, **kwargs
)
Он может либо обернуть существующий тензор (передать аргумент input_tensor), либо создать тензор-заполнитель (передать аргументы input_shape, и необязательно dtype).
В целом рекомендуется использовать функциональный API слоёв через Input, (что создаёт InputLayer) без прямого использования InputLayer.
При использовании InputLayer с моделью Keras Sequential, его можно пропустить, перенеся параметр input_shape в первый слой после InputLayer.
Этот класс может создавать заполнители для tf.Tensors, tf.SparseTensors и tf.RaggedTensors, выбрав 'sparse=True' или 'ragged=True'. Обратите внимание, что 'sparse' и 'ragged' не могут быть одновременно установлены в значение True. Пример использования:
# With explicit InputLayer.
model = tf.keras.Sequential([
tf.keras.layers.InputLayer(input_shape=(4,)),
tf.keras.layers.Dense(8)])
model.compile(tf.optimizers.RMSprop(0.001), loss='mse')
model.fit(np.zeros((10, 4)),
np.ones((10, 8)))
# Without InputLayer and let the first layer to have the input_shape.
# Keras will add a input for the model behind the scene.
model = tf.keras.Sequential([
tf.keras.layers.Dense(8, input_shape=(4,))])
model.compile(tf.optimizers.RMSprop(0.001), loss='mse')
model.fit(np.zeros((10, 4)),
np.ones((10, 8)))
| Аргументы | |
|---|---|
input_shape | Кортеж формы (без учёта оси пакетности), или экземпляр TensorShape (без учёта оси пакетности). |
batch_size | Необязательный размер пакета входных данных (целое число или None). |
dtype | Необязательный тип данных входных данных. Если не указан, будет использован стандартный тип с плавающей точкой Keras. |
input_tensor | Необязательный тензор для использования в качестве входных данных слоя. Если установлен, слой будет использовать tf.TypeSpec этого тензора вместо создания нового тензора-заполнителя. |
sparse | Булево значение, указывающее, должен ли создаваемый заполнитель быть разреженным. По умолчанию False. |
ragged | Булево значение, указывающее, должен ли создаваемый заполнитель быть разрозненным. В этом случае значения 'None' в аргументе 'shape' обозначают разрозненные измерения. Для получения дополнительной информации о RaggedTensors см. это руководство. По умолчанию False. |
name | Необязательное имя слоя (строка). |
© 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.4/api_docs/python/tf/keras/layers/InputLayer