KNN CNN 識別手寫數字 核心代碼理解

 拓展維度

# f1測試數據輸入維度擴展 由5*784==>5*1*784
# 5 500 784 (3D) 5張測試圖片 500張訓練圖片 結果有 5* 500 *784種組合, 拓展維度是爲了訓練圖片和測試圖片之間計算差值
f1 = tf.expand_dims(testDataInput,1)

with tf.Session() as sess:
    print('tensorflow運行結果------------')
    p1 = sess.run(f1,feed_dict={testDataInput:testData[0:5]})
    print('p1維度=',p1.shape) #(5,1,784)

輸出結果
tensorflow運行結果------------
p1維度= (5, 1, 784)