在Keras中,math_ops.floor等效

问题描述 投票:2回答:1

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

tensorflow keras keras-layer floor
1个回答
3
投票

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

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

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

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

© www.soinside.com 2019 - 2024. All rights reserved.