keras.layers.add()和keras.layer.conatenate()

keras.layers.add()和keras.layer.conatenate()

add对张量执行求和运算
concatenate对张量进行串联运算

在深度神经网络中,经常会遇到需要把张量结合在一起的情况,比如Inception网络。add()和conetenate()经常出现,用来将两个张量结合在一起。
那么这两个函数有什么区别呢?

add():直接对张量求和
例如:

import 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)
added = keras.layers.add([x1, x2])

out = keras.layers.Dense(4)(added)
model = keras.models.Model(inputs=[input1, input2], outputs=out)
model.summary()

输出结果为:

keras.layers.add()和keras.layer.conatenate()_第1张图片
add层将dense_1层的输入和dense_2层的输入加在了一起,是张量元素内容相加。

conatenate():串联一个列表的输入张量。
例如:

import 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)
conatenated = keras.layers.conatenate([x1, x2])

out = keras.layers.Dense(4)(conatenated)
model = keras.models.Model(inputs=[input1, input2], outputs=out)
model.summary()

输出结果为:

keras.layers.add()和keras.layer.conatenate()_第2张图片
可以看到conatenate对最后一维进行了串联,通道数变成了8+8=16,可以指指定axis=x来指定空间的第x维串联。

总结:
add对张量执行求和运算
concatenate对张量进行串联运算

你可能感兴趣的:(学习)