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)