有什麼方法可以在keras層做比較嗎？

Question

我想比較keras模型中的兩個數字。該層的輸入是張量變化的，並且該層將該張量變量與常數進行比較。然後它會返回0或1. 有什麼方法嗎？我試圖在theano中找到一個功能來完成這項工作，但失敗了。

Answer 1

您可以找到功能keras backend

import keras.backend as K 

你需要的是其中之一：K.equal，K.greater，K.greater_equal等

可以使用Lambda層：

Lambda(lambda x: K.cast(K.greater_equal(x,constant),'float32'),output_shape=sameAsInputShape)