自定义激活功能以进行推理

Question

我正在尝试在Tensorflow中创建自己的自定义激活功能。这主要是因为我必须测试一些S型近似值。问题是我只需要推理即可。我在帖子here中找到了此解决方案。但是，我只想将其用于推理，因此我觉得它毫无意义地过于复杂。

对于任何建议，我将不胜感激。