tensorflow - Keras 后端中的逻辑 AND/OR

Question

Tensorflow 有 tf.逻辑_and() 和 tf.逻辑_or() 用于比较两个 bool 张量，即 tf.逻辑_and(x,y)= =TRUE 如果x==TRUE 且 y==TRUE (doc) 。但我在 Keras 后端找不到类似的东西。它们有 keras.backend.any() 和 .all() ，但这用于张量内的聚合，而不是张量之间的聚合。我一直不得不使用嵌套 K.switch() 函数的解决方法，但它非常不优雅。

Answer 1

让x和y是相同形状的 bool keras 张量。

要进行元素或，请执行以下操作:

keras.backend.any(keras.backend.stack([x, y], axis=0), axis=0)

要按元素求“和”，请执行以下操作:

keras.backend.all(keras.backend.stack([x, y], axis=0), axis=0)

这里keras.backend.stack([x, y], axis=0)将 x 和 y 堆叠到一个新张量中，附加维度为 0。之后 keras.backend.any沿新维度取逻辑或，并且 keras.backend.any取逻辑与。