在Keras中添加自定义激活功能

Question

我正在尝试在Keras中实现自定义激活功能 该功能如下

def picewise(x):
   if x > 0.5 :
      return 1
   elif if x < 0.5:
      return 0
   else:
      return x + 0.5

我将激活功能称为

model.add(Dense(128,activation = picewise))

但我收到了一大堆错误。如何在Keras

中成功实现上述工作激活功能

Answer 1

此处x不是原始类型。 x是一个张量。因此，您尝试实现激活功能的方式将不起作用。您可能需要以下内容：

import tensorflow as tf
def picewise(x):
    z = tf.where(x >= 0.5, x - x + 1.0 , x)
    y = tf.where(z <= -0.5,z -  z + 0, z + 0.5)
    return y