Python tf.keras.layers.Subtract用法及代碼示例

減去兩個輸入的層。

繼承自：Layer，Module

用法

tf.keras.layers.Subtract(
    **kwargs
)

它將大小為 2、形狀相同的張量列表作為輸入，並返回同樣形狀的單個張量 (inputs[0] - inputs[1])。

例子：

import keras

    input1 = keras.layers.Input(shape=(16,))
    x1 = keras.layers.Dense(8, activation='relu')(input1)
    input2 = keras.layers.Input(shape=(32,))
    x2 = keras.layers.Dense(8, activation='relu')(input2)
    # Equivalent to subtracted = keras.layers.subtract([x1, x2])
    subtracted = keras.layers.Subtract()([x1, x2])

    out = keras.layers.Dense(4)(subtracted)
    model = keras.models.Model(inputs=[input1, input2], outputs=out)