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

Multiply 層的函數接口。

用法

tf.keras.layers.multiply(
    inputs, **kwargs
)

例子：

x1 = np.arange(3.0)
x2 = np.arange(3.0)
tf.keras.layers.multiply([x1, x2])
<tf.Tensor:shape=(3,), dtype=float32, numpy=array([0., 1., 4.], ...)>

在函數模型中的用法：

input1 = tf.keras.layers.Input(shape=(16,))
x1 = tf.keras.layers.Dense(8, activation='relu')(input1) #shape=(None, 8)
input2 = tf.keras.layers.Input(shape=(32,))
x2 = tf.keras.layers.Dense(8, activation='relu')(input2) #shape=(None, 8)
out = tf.keras.layers.multiply([x1,x2]) #shape=(None, 8)
out = tf.keras.layers.Dense(4)(out)
model = tf.keras.models.Model(inputs=[input1, input2], outputs=out)