报错截图
问题出自于我的custom layer。Keras在初始化custom layer的时候,必须加上compute_output_shape函数才能自动判断input_shape。
否则的话,如果实例化了两个custom layer: L_1和L_2。在不加上述函数的情况下,L_2层的input_shape会默认和L_1层相同。
在keras自己的文档上其实说的很清楚了,但是比较容易忽视。以下是链接
https://keras.io/zh/layers/writing-your-own-keras-layers/
另一个博客我觉得对理解这个问题也有一些帮助
https://www.cnblogs.com/yaos/p/12128153.html
不知道引用格式对不对:D