xs = tf.placeholder(tf.float32, [None, 784], name='x_input') x_image = tf.reshape(xs, [-1, 28, 28, 1]) 中的reshape起到了什么作用?
时间: 2024-05-27 12:12:30 浏览: 78
使用vue.js2.0 + ElementUI开发后台管理系统详细教程(二)
reshape的作用是将输入的数据xs按照给定的形状[-1, 28, 28, 1]进行重新排列,并返回新的tensor x_image。在这个例子中,reshape函数的参数中[-1, 28, 28, 1]表示将xs分成若干个28×28大小的图片,每个图片有1个通道。这个重新排列后的tensor可以作为神经网络的输入。
阅读全文