tensorflow - 创建像 relu6 这样的自定义非线性

Question

有人可以帮我找到允许我修复阈值（最大值）不超过 relu6 的函数。

我试过了

 X = max(X , 6) 

但我收到了这个错误：

（OperatorNotAllowedInGraphError：Graph 执行中不允许使用 atf.Tensor作为 Python 。使用 Eager 执行或使用 @tf.function 装饰此函数）bool

. 先感谢您

Answer 1

您可能想尝试：

X = tf.math.maximum(X,6)

详细信息：https ://www.tensorflow.org/api_docs/python/tf/math/maximum

我希望这有帮助。