TensorFlow 中的layers 模組提供了高階API，使我們能輕易建立出神經網路。包括全連線層，卷積層，啟用函式，dropout regularization。在本文中，將構建CNN網路來進行手寫數字識別。

MNIST資料集包括60000個訓練樣本，10000個測試樣本，每個樣本為28*28畫素的圖片。

本例中使用的網路結構如下：

卷積層1：32個5*5 filter，啟用函式為ReLU

池化層1：2*2，步長為2

卷積層2: 64個5*5 filter，啟用函式為ReLU

池化層2: 2*2，步長為2

Dense Layer 1（全連線層）：1024個neurons，dropout regularization rate 為 0.4

API中的三個函式(這三個函式的輸入輸出均為tensor)：tf.layers.conv2d(), tf.layers.max_pooling2d(), tf.layers.dense()

首先給出定義模型的程式碼：

下面對程式碼細節進行解釋：

Input Layer:

對於一個二維的影像，首先將資料轉化為以下形式[batch_size,image_width,image_height,channels]

其中batch_size = -1，表示生成自適應大小的tensor。

Convolutional Layer:

此處值得注意的是，padding，padding有兩個可選值 valid和same，預設為valid，當padding值為same值時，卷積是會在邊緣處補0，使得輸出tensor的width*height與輸入相同。

Dense Layer：

這裡的全連線層包含1024個神經元，在連線至全連線層之前，我們首先將原有的tensor轉換。由pool2輸出的tensor（[batch_size,image_width,image_height,channels]）轉換為二維的tensor（[batch_size,features]）。Features需要自己計算出來，因為pool2輸出tensor的格式為[batch_size,7,7,64]。

為了防止過擬合，需要使用drop regularization，

這裡的引數training是一個Boolean值，只有在訓練階段，才需要使用dropout。當training = true時，這一層才會執行。輸出為[batch_size, 1024]

Logits Layer:

這裡使用了預設的啟用函式linear activation

Generate Predictions：

Logits Layer輸出一個tensor[batch_size,10](類似一個batch_size 行，10列的一個矩陣，對應每一張圖片，都會產生一個向量，每個向量有10個值),在此，我們可以生成兩種型別的返回值，（1）返回0,1,2,3,4,5,6,7,8,9中的一個數字（2）返回對應每個數字的機率，例如0的機率是0.2,1的機率是0.7等等。

在本例中，我們返回向量（10個值）中，最大值對應的類別。使用tf.argmax()獲得對應的索引。