Keras 中的 math_ops.floor 等效项

Question

我正在尝试在 Keras 中实现一个自定义层，我需要将浮点数张量转换[a, 1+a)为二进制张量以进行屏蔽。我可以看到 Tensorflow 有一个floor函数可以做到这一点，但 Keras 似乎没有在keras.backend. 知道我该怎么做吗？

Answer 1

根据 OP 的要求，我将提及我在评论中给出的答案并详细说明：

简短回答：如果您使用tf.floor().

长答案：keras.backend.*在 1) 需要预处理或增加传递给 Tensorflow 或 Theano 后端的实际函数的参数或对返回的结果进行后处理的情况下，使用 Keras 后端函数（即）是必要的。例如，mean后端的方法也可以使用布尔张量作为输入，但是reduce_meanTF中的方法需要数字类型作为输入；或 2) 您想编写一个适用于所有 Keras 支持的后端的模型。

否则，直接使用大部分真正的后端功能就可以了；但是，如果该功能已在keras.backend模块中定义，则建议改用该功能。