Keras自定义层如何获得多重回报

Question

我正在使用Tensorflow在Keras中编写自定义图层。调用函数有多个步骤，如果要满足条件，我想知道是否可以跳过某些步骤。 例如，让我们使用以下功能：

def test_fn(x):
    if x < 10:
        return 0
    else:
        return 1

在Keras中有没有办法做类似的事情？ 我有这样的东西（简体）：

class Test_layer(layers.Layer):

    def call(self, inputs, **kwargs):
        test = get_test_fn(inputs)
        if test < 10:
            return 0
        else:
            return 1

我知道那里的语法是错误的，但是有没有办法做一些等效的事情？