将共享层的输出传递给层
作者:互联网
我正在解决一个问题,我必须创建一个相同共享层的列表并将其传递给另一层.所以我使用了for循环并添加到列表中.然后,我必须将那些推数传递到另一层.列表无法传递到另一层.如何执行呢?
x = Input(shape=...)
shared_layer = MySharedLayer(...)
outputs=[]
# Step 2: Iterate for Ty steps
for t in range(Ty):
out = output_layer(x)
outputs.append(out)
lastlayer=Dense(shape=...)(outputs)
解决方法:
您可以简单地使用keras.layers.Concatenate,连接层将输入列表连接起来.
x = Input(shape=...)
shared_layer = MySharedLayer(...)
outputs=[]
# Step 2: Iterate for Ty steps
for t in range(Ty):
out = output_layer(x)
outputs.append(out)
concat = concatenate(outputs)
lastlayer=Dense(shape=...)(concat)
标签:keras,machine-learning,deep-learning,python 来源: https://codeday.me/bug/20191210/2104604.html