其他分享
首页 > 其他分享> > 将共享层的输出传递给层

将共享层的输出传递给层

作者:互联网

我正在解决一个问题,我必须创建一个相同共享层的列表并将其传递给另一层.所以我使用了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