Python 张量流:张量的反向展平

Python 张量流:张量的反向展平,python,tensorflow,flatten,Python,Tensorflow,Flatten,OS:Kubuntu 14.04,Tensorflow版本=0.12,Python版本=2.7 在我的最后一层之后,我的神经网络输出的形状是[batch,height,width,2],在我的例子中是[64,32,256,2] 之后我用 output = tf.contrib.layers.flatten(input=output) 它产生一个形状为[6416384] 出于评估目的,我想在另一个函数中反转此展平,以获得原始的[64,32,256,2]张量 你会怎么做 简单地重塑到你想要的形状:

OS:Kubuntu 14.04,Tensorflow版本=0.12,Python版本=2.7

在我的最后一层之后,我的神经网络输出的形状是
[batch,height,width,2]
,在我的例子中是
[64,32,256,2]

之后我用

output = tf.contrib.layers.flatten(input=output)
它产生一个形状为[6416384]

出于评估目的,我想在另一个函数中反转此展平,以获得原始的
[64,32,256,2]
张量


你会怎么做

简单地
重塑
到你想要的形状:
tf.reforme(输出[-1,32,256,2])

你试过
tf.reforme(输出[-1,32,256,4])
吗?是的,这比我想象的要容易得多。顺便说一句,必须有一个2而不是4,因为我的第四维是2。如果你把那写作为答案,我会记为正确