Keras 2 API 文档 / 层 API / 合并层 / Subtract 层

减法层

[源代码]

Subtract

tf_keras.layers.Subtract(**kwargs)

Subtracts two inputs 的层。

它接收一个包含两个张量的列表作为输入,这两个张量具有相同的形状,并返回一个单一的张量 (inputs[0] - inputs[1]),形状也相同。

示例

    import tf_keras.src as 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)