在本地和云(kaggle)上，当尝试拟合Keras模型时，笔记本会死亡(忘记任何导入或变量，停止执行单元格)。此问题仅在架构中存在自定义层SingularityExtractor2D时出现。
你可以在这里找到笔记本:Github
自定义回调GateOfLearning已经过多次测试，每次在任何模型架构下都能完美地工作。笔记本电脑已经在GPU和CPU上运行了，问题仍然存在。
当调用tf.ones函数时，形状必须是(spatial_0, ..., spatialN-1, in_channels, out_channels)，而不是调用(batch_size, spatial_0, ..., spatial_N-1, channels)时该层得到的张量的形状