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

Question

我根据DEEP MNIST专家教程构建了一个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, ...)