训练或预测时执行Tensorflow图

时间:2019-06-17 09:37:57

标签: tensorflow machine-learning neural-network

我基本上是想找到训练模型并分别对模型进行预测时执行的计算图。

我尝试使用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 ++代码。

0 个答案:

没有答案