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

08-24 21:05
查看更多