tensorflow 错误: Using a `tf.Tensor` as a Python `bool` is not allowed

Question

我正在努力实现 tensorflow 中的激活函数在Python中。

代码如下:

def myfunc(x):
    if (x > 0):
        return 1
    return 0

但我总是收到错误:

Using a tf.Tensor as a Python bool is not allowed. Use if t is not None:

Answer 1

使用tf.cond :

tf.cond(tf.greater(x, 0), lambda: 1, lambda: 0)

另一种解决方案，另外还支持多维张量:

tf.sign(tf.maximum(x, 0))

但请注意，此激活的梯度在各处均为零，因此神经网络不会从中学习任何内容。