Keras中的math_ops.floor等效项

时间:2018-09-05 11:51:37

标签: tensorflow keras keras-layer floor

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

1 个答案:

答案 0 :(得分:3)

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

简短答案:如果您使用tf.floor(),则不会遇到任何重大问题。

长答案:在以下情况下,必须使用Keras后端函数(即keras.backend.*):1)需要预处理或扩充传递给Tensorflow或Theano后端的实际功能或对返回的结果进行后处理。例如,mean方法就是这种情况的一个例子。或2)您想编写一个适用于所有Keras支持的后端的模型。

否则,直接使用大多数真实的后端函数就可以了;但是,如果功能是在keras.backend模块中定义的,则建议改用该功能。