x = tf.reshape(x, [x_size[1], x_size[0], x_size[2], x_size[3]])