Tensorflow：张量的反向展平

Question

操作系统：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]恢复原始张量。

那怎么办呢？

Answer 1

简单reshape到你想要的形状：tf.reshape(output, [-1, 32, 256, 2])。