Keras建模为tensorflow.keras

Question

我有一个使用Keras（使用Tensorflow作为后端）制作的模型。 现在，我想在Keras版本中使用Tensorflow。

但是，替换此行

from keras.engine.topology import get_source_inputs

与此行

from tensorflow.keras.engine.topology import get_source_inputs

不起作用。它说

ImportError: No module named 'tensorflow.keras.engine'

在get_source_inputs的哪里可以找到Tensorflow函数？

Answer 1

将Keras更新到2.2.4版似乎对我有用！

然后可以通过以下方式之一导入get_source_inputs函数：

from keras.utils import get_source_inputs
from keras.utils.layer_utils import get_source_inputs

或者如果您不是将Keras用作外部库而是TF版本，则可以使用以下命令导入该函数：

from tensorflow.keras.utils import get_source_inputs

希望这会有所帮助！