当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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

Multiply 层的函数接口。

用法

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

参数

  • inputs 输入张量列表(至少 2 个)。
  • **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)

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.layers.multiply。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。