我正在尝试在Keras中实现自定义层,需要将浮点[a, 1+a)
的张量转换为二进制张量以进行掩膜。我可以看到Tensorflow有一个floor
函数可以做到这一点,但是Keras在keras.backend
中似乎没有。知道我该怎么做吗?
答案 0 :(得分:3)
根据OP的要求,我将在评论中提及我给出的答案,并详细说明:
简短答案:如果您使用tf.floor()
,则不会遇到任何重大问题。
长答案:在以下情况下,必须使用Keras后端函数(即keras.backend.*
):1)需要预处理或扩充传递给Tensorflow或Theano后端的实际功能或对返回的结果进行后处理。例如,mean
方法就是这种情况的一个例子。或2)您想编写一个适用于所有Keras支持的后端的模型。
否则,直接使用大多数真实的后端函数就可以了;但是,如果功能是在keras.backend
模块中定义的,则建议改用该功能。