如何将Conv2D层作为自定义层传递给CoreML？

Question

我阅读了有关自定义图层的this blog post，并希望在Metal中重新实现Conv2D。由于某种原因，我无法使用that script在Keras中使用Lambda创建自定义Conv2D图层。

这是代码和调试输出：

• https://www.diffchecker.com/GQRyLddR（代码）
• https://www.diffchecker.com/aJQJAfJc（调试）