W_fc1 = weight_variable([7*7*64, 1024]) b_fc1 = bias_variable([1024]) h_pool2_flat = tf.reshape(h_pool2, [-1, 7*7*64]) h_fc1 = tf.nn.relu(tf.matmul(h_pool2_flat, W_fc1) + b_fc1)
第1全結合層では、畳み込み層とプーリング層においては、データを4次元で処理していたが、実際はどれに分類される画像であるか判断するために、画像全体をまとめて処理する必要があるため、1次元のベクトルに直す処理を行っている。