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