tf.keras.layers.InputLayer
| Просмотреть исходный код на GitHub |
Слой, используемый в качестве точки входа в сеть (граф слоёв).
tf.keras.layers.InputLayer(
input_shape=None,
batch_size=None,
dtype=None,
input_tensor=None,
sparse=None,
name=None,
ragged=None,
type_spec=None,
**kwargs
)
Он может либо обернуть существующий тензор (передать аргумент input_tensor), либо создать тензор-заполнитель (передать аргументы input_shape, и необязательно, dtype).
В целом рекомендуется использовать функциональную модель Keras через 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 | Необязательный тип данных входных данных. Если не указан, будет использован стандартный тип float Keras. |
input_tensor | Необязательный тензор для использования в качестве входных данных слоя. Если задан, слой будет использовать tf.TypeSpec этого тензора, а не создавать новый тензор-заполнитель. |
sparse | Булево значение, указывает ли создаваемый заполнитель на разреженный тензор. По умолчанию False. |
ragged | Булево значение, указывает ли создаваемый заполнитель на нерегулярный тензор. В этом случае значения None в аргументе shape представляют нерегулярные измерения. Для получения дополнительной информации о tf.RaggedTensor см. это руководство. По умолчанию False. |
type_spec | Объект tf.TypeSpec для создания Input. Этот tf.TypeSpec представляет весь пакет. Если указан, все другие аргументы, кроме name, должны быть None. |
name | Необязательное имя слоя (строка). |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/InputLayer