Keras 3 API 文档 / 层 API / 合并层 / 减法层

减法层

[源代码]

Subtract

keras.layers.Subtract(**kwargs)

执行逐元素减法。

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

示例

>>> input_shape = (2, 3, 4)
>>> x1 = np.random.rand(*input_shape)
>>> x2 = np.random.rand(*input_shape)
>>> y = keras.layers.Subtract()([x1, x2])

在 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)