keras.get_ssion().graph在tensorflow2.x中不起作用

Question

我可以通过tensorflow1.x中的以下代码绘制keras模型的图形

from tensorflow.python.keras import backend as K
model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation='relu'),
    keras.layers.Dense(10)
])
...
graph=K.get_session().graph


graph_def=graph.as_graph_def()
print(graph_def)

​

​

但是，当我将tensorflow版本更改为2.x时，它不起作用。我在tensorflow2.1中得到了类似下面图片的结果。

​

​

如何在tensorflow2.x中获取keras模型的图表？

Answer 1

Tensorflow迁移指南将是您可以开始的地方-

Migrate your TensorFlow 1 code to TensorFlow 2