是否有任何方法可以在keras层中进行比较？

我想比较keras模型中的两个数字。该层的输入是一个张量变量，该层将该张量变量与常数进行比较。然后它将返回0或1。

---

有什么办法吗？我试图在中找到一个函数来执行此任务，但失败。

您可以在中找到函数

你需要的是其中之一：K.equal，K.greater，K.greater\u equal，等等

您可以使用

Lambda

层：

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

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