0
케라 모델에서 두 개의 숫자를 비교하고 싶습니다. 이 레이어의 입력은 텐서 변수이며이 레이어는이 텐서 변수를 상수와 비교합니다. 그런 다음 0 또는 1을 반환합니다. 방법이 있습니까? 나는이 일을하기 위해 theano에서 기능을 찾으려고했지만 실패했다.케라 레이어에서 비교할 수있는 방법이 있습니까?
A
답변
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)