Question

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

Answer 1

您可以简单地使用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)

将共享层的输出传递到层

1 个答案: