如何连接最后一轴以外的角膜层

Thi*_*ill 5 keras tensorflow

我试图将keras层连接到最后一个轴之外。

concat_layer = keras.layers.concatenate([layer1,layer2],axis=3);
Run Code Online (Sandbox Code Playgroud)

layer1和layer2的形状都是(?,7,7),现在我希望它变成(?,7,7,2)而不是(?,7,14)。如果我写的像axis = 3,它将返回“ IndexError:列表分配索引超出范围” ...

我该怎么办?非常感谢!

P-G*_*-Gn 1

您可以使用keras.backend.stack,例如:

keras.backend.stack([layer1,layer2],axis=-1)
Run Code Online (Sandbox Code Playgroud)