Existe-t-il une méthode pour faire des comparaisons dans les calques keras?

Question

Je veux comparer deux nombres dans le modèle de keras. L'entrée de cette couche est une variable tensorielle et cette couche compare cette variable tensorielle avec une constante. Ensuite, il retournera 0 ou 1. Y a-t-il une méthode? J'ai essayé de trouver une fonction à Theano pour faire ce travail mais j'ai échoué.

Answer 1

Vous pouvez trouver les fonctions keras backend

import keras.backend as K 

Qu'est-ce que vous avez besoin est un de ces: K.equal, K.greater, K.greater_equal, etc.

Vous pouvez utiliser un Lambda couche pour cela:

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