tf.keras.layers.Add
| Просмотреть исходный код на GitHub |
Слой, который складывает список входных данных.
tf.keras.layers.Add(
**kwargs
)
Он принимает на вход список тензоров, все с одинаковой формой, и возвращает один тензор (также с одинаковой формой).
Примеры:
input_shape = (2, 3, 4) x1 = tf.random.normal(input_shape) x2 = tf.random.normal(input_shape) y = tf.keras.layers.Add()([x1, x2]) print(y.shape) (2, 3, 4)
Используется в функциональной модели:
input1 = tf.keras.layers.Input(shape=(16,)) x1 = tf.keras.layers.Dense(8, activation='relu')(input1) input2 = tf.keras.layers.Input(shape=(32,)) x2 = tf.keras.layers.Dense(8, activation='relu')(input2) # equivalent to `added = tf.keras.layers.add([x1, x2])` added = tf.keras.layers.Add()([x1, x2]) out = tf.keras.layers.Dense(4)(added) model = tf.keras.models.Model(inputs=[input1, input2], outputs=out)
| Аргументы | |
|---|---|
**kwargs | стандартные ключевые слова аргументов слоя. |
© 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/Add