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)