我基本上是想找到训练模型并分别对模型进行预测时执行的计算图。
我尝试使用tf.get_default_graph(),但是我不确定应该得到什么。
如果我在代码中的各个位置检索到的操作数都不同。
建立模型后。 编译模型后。 拟合模型后。 使用模型进行预测之后。 拟合后,然后使用模型进行预测。
model = Sequential()
model.add(Dense(output_shape, input_shape=input_shape, activation=None))
# Option 1
model.compile(optimizer="sgd", loss="binary_crossentropy")
# Option 2
model.fit(x=x_train, y=y_train, batch_size=4, epochs=2, verbose=1)
# Option 3
model.predict(x_train)
# Option 4
在查看张量流跟踪时,我注意到图中存在许多跟踪中缺少的操作。特别是在预测中,计算图显示约80个操作,而轨迹仅显示10个。
我试图查看tensorflow源代码以查找实际执行的内容,但是我已经陷入僵局,需要研究核心c ++代码。