How do I fix a dimension error in TensorFlow?,"x_image = tf.reshape(tf_in, [-1, 2, 4, 1])"