ケラス層で比較する方法はありますか？

Question

ケラスモデルで2つの数字を比較したいと思います。この層の入力はテンソル変数であり、この層はこのテンソル変数を定数と比較します。その後、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)