如何在TensorFlow中测试各个图层？

Question

我根据DEEP MNIST Expert教程构建了一个7层卷积网络。 我又添加了两个卷积层。

一切运行良好，但我想尝试将1024 x 10数组直接输入到完全连接的层中，并规避卷积层。

没有重建整个网络的方法吗？

Answer 1

在卷积层和完全连接层之间，为完全连接层的输入创建一个占位符： input_to_fc = tf.placeholder_with_default(previous_layer, shape=(None, 1024*10)) 。 您可以通过将输入直接输入到input_to_fc张量来绕过卷积层。

例：

...
conv = tf.layers.conv2d(...)
flatten = tf.layers.flatten(...)
input_to_fc = tf.placeholder_with_default(flatten, shape=OUTPUT_SHAPE_OF_PREVIOUS_LAYER))
fc = tf.layers.dense(input_to_fc, ...)