是否有任何方法可以在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)