tensorflow：flatten batch

tensorflow中的tensor的shape變換

如將一個（？，128，160，6）的tensor轉換（flatten）爲（？，128×160×6）的shape

shape = x.get_shape().as_list()        # a list: [？，128，160，6]
dim = numpy.prod(shape[1:])            # dim = prod(128，160，6)=128×160×6
x2 = tf.reshape(x, [-1, dim])           # -1 means "all"

這樣，x2的shape爲（？，128×160×6）