tf.keras.layers.InputLayer
| Просмотреть исходный код на GitHub |
Слой, используемый в качестве точки входа в сеть (граф слоёв).
Наследуется от: Layer
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 | Необязательный тип данных входных данных. Если не указано, используется стандартный тип float Keras. |
input_tensor | Необязательный тензор для использования в качестве входных данных слоя вместо создания заполнителя. |
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.3/api_docs/python/tf/keras/layers/InputLayer