tf.nn.conv2d注意事项
2018-07-16 本文已影响7人
yalesaleng
例如：
Output = tf.nn.conv2d(input, weight, strides = [1,1,1,1], padding = ‘SAME’]
一定要注意input是一个4-D数据，
所以如果你输入的是一副3-D(例如 4806403)图像，
那么要用tf.reshape来将input扩展到4-D。
具体code：
Input = tf.reshape(input, shape = [1,480,640,3])
