终于可以在tf 2.0上自定义激活函数和损失函数了。当时遇到了好几个坑。
参考的Stackoverflow上的答案。