我试图将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:列表分配索引超出范围” ...
我该怎么办?非常感谢!
您可以使用keras.backend.stack,例如:
keras.backend.stack([layer1,layer2],axis=-1)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
1230 次 |
| 最近记录: |