美文网首页
tf.keras.layers.Add

tf.keras.layers.Add

作者: 又双叒叕苟了一天 | 来源:发表于2021-01-01 12:59 被阅读0次

    对input list按元素求和, 输出的shape和输入的shape相同.

    示例:

    input_shape = (2, 3, 4)
    x1 = tf.random.normal(input_shape)
    x2 = tf.random.normal(input_shape)
    y = tf.keras.layers.Add()([x1, x2])
    print(y.shape)
    
    input1 = tf.keras.layers.Input(shape=(16,))
    x1 = tf.keras.layers.Dense(8, activation='relu')(input1)
    input2 = tf.keras.layers.Input(shape=(32,))
    x2 = tf.keras.layers.Dense(8, activation='relu')(input2)
    # equivalent to `added = tf.keras.layers.add([x1, x2])`
    added = tf.keras.layers.Add()([x1, x2])
    out = tf.keras.layers.Dense(4)(added)
    model = tf.keras.models.Model(inputs=[input1, input2], outputs=out)
    

    相关文章

      网友评论

          本文标题:tf.keras.layers.Add

          本文链接:https://www.haomeiwen.com/subject/jpkjoktx.html