Gibt es irgendeine Methode, in Keras-Schichten zu vergleichen?

Question

Ich möchte zwei Zahlen im Keras-Modell vergleichen. Die Eingabe dieser Schicht ist eine Tensorvariable und diese Schicht vergleicht diese Tensorvariable mit einer Konstanten. Dann gibt es 0 oder 1 zurück. Gibt es eine Methode? Ich habe versucht, eine Funktion in theano zu finden, um diese Aufgabe zu erledigen, aber es ist fehlgeschlagen.

Answer 1

Sie können die Funktionen finden in keras backend

import keras.backend as K 

Was Sie brauchen, ist eine davon: K.equal, K.greater, K.greater_equal usw.

Sie können ein verwenden Lambda Schicht dafür:

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