python之Tensorflow:张量的反向展平

操作系统:Kubuntu 14.04，Tensorflow 版本 = 0.12，Python 版本 = 2.7

在我的最后一层之后，我的神经网络的输出具有形状[batch, height, width, 2]，即 [64, 32, 256, 2]就我而言。

然后我用

output = tf.contrib.layers.flatten(input=output) 

生成一个形状为[64, 16384]

出于评估目的，我想在不同的函数中反转这种展平以获得原始的 [64, 32, 256, 2] 张量。

如何做到这一点？

请您参考如下方法:

只需reshape到你想要的形状:tf.reshape(output, [-1, 32, 256, 2])。