我想在keras model中加入自定義function，功能是將tensor中每一個element和一個數比大小，若大於此數則輸出1，若小於此數則輸出0，因此我define了一個function：

def Compare(input_tensor):
    input_shape = input_tensor.get_shape().as_list()
    y = tf.placeholder(dtype=tf.float32, shape=[None,55,55,96], name='y')

    for j in range(input_shape[1]):
        for k in range(input_shape[2]):
            for z in range(input_shape[3]):
                y = tf.map_fn(lambda x: tf.cond(tf.greater(x[j][k][z], -4.4), lambda: 1.0, lambda: 0.0),input_tensor,dtype=tf.float32)
    print(y)
    return y

其中，[55,55,96]是我的input tensor shape也是output tensor shape，-4.4是我用來比大小的門檻值，如上面所說，大於-4.4則輸出1，小於-4.4則輸出0。
接這我再將此function用keras.layers.lambda轉換成layer並放到model裡面

conv1     = Conv2D(96,(11,11),strides=4,activation=None)(input_img)
conv1     = Lambda(Compare)(conv1)

我是用jupyter notebook，沒有出現error但一直跑不完，我不知道是不是我的code寫得不好或有其他問題，希望各位邦友能給我一些建議，謝謝大家！