如何在TensorFlow中用Python语言制作分段激活函数？

import tensorflow as tf

def piecewise_activation(x):
    condition1 = tf.less(x, -1)  # 定义第一个分段的条件
    condition2 = tf.logical_and(tf.greater_equal(x, -1), tf.less(x, 0))  # 定义第二个分段的条件
    condition3 = tf.logical_and(tf.greater_equal(x, 0), tf.less(x, 1))  # 定义第三个分段的条件
    condition4 = tf.greater_equal(x, 1)  # 定义第四个分段的条件

    segment1 = tf.multiply(tf.add(x, 1), 0.5)  # 第一个分段的计算
    segment2 = tf.multiply(tf.add(x, 1), 0.5)  # 第二个分段的计算
    segment3 = tf.multiply(tf.subtract(1, x), 0.5)  # 第三个分段的计算
    segment4 = tf.multiply(tf.subtract(1, x), 0.5)  # 第四个分段的计算

    output = tf.where(condition1, segment1, tf.where(condition2, segment2, tf.where(condition3, segment3, segment4)))
    return output

x = tf.placeholder(tf.float32, shape=[None, input_size])  # 定义输入张量
y = piecewise_activation(x)  # 使用分段激活函数